HashSlap-Summer-of-Code/ml-core
Implement Modular AutoML System for Model Selection and Tuning
オープン
#12 opened on 2025/06/18
Advancedenhancementhacktoberfesthssoc
Repository metrics
- Stars
- (3 個のスター)
- PR merge metrics
- (PR metrics pending)
説明
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.