HashSlap-Summer-of-Code/ml-core
Implement Modular AutoML System for Model Selection and Tuning
开放
#12 创建于 2025年6月18日
Advancedenhancementhacktoberfesthssoc
仓库指标
- 星标
- (3 个星标)
- PR 合并指标
- (PR 指标待抓取)
描述
Description:
Design a lightweight AutoML system that automates model selection and hyperparameter tuning for supported algorithms (e.g., Perceptron, KNN, Decision Tree). It should be terminal-based, configurable, and able to produce ranked results with visualizations.
Expected Tasks:
- Create a Python script
automl.pyinside a newtools/orautoml/folder. - Define a consistent interface for models with
.fit()and.predict()methods. - Implement support for ingesting CSV datasets (with
pandas). - Include multiple algorithms from the repo (at least 4).
- Add a configuration for hyperparameter tuning using grid or random search.
- Train each model using cross-validation and evaluate with metrics like accuracy, F1-score, or RMSE.
- Log results to a CSV file and display the best-performing models.
- Use
matplotliborseabornto generate performance plots. - Add CLI arguments to control dataset path, model list, metric, and tuning strategy.
- Document setup and usage in the root
README.mdand add an example dataset.
Stretch Tasks:
- Add time or resource constraints to avoid slow models dominating.
- Support for classification vs regression auto-detection.
- Optional visual dashboard using
streamlitorgradio.