COSC481W-2024Fall / MLite

A user-friendly ML platform that lets non-experts upload datasets, receive model suggestions, schedule training, and deploy models for real-time predictions
0 stars 0 forks source link

ML - Training Script - Decision Trees #8

Closed Mohammad4844 closed 1 month ago

Mohammad4844 commented 1 month ago

User Story

As a user, I would like to train a decision tree model. I want to be able to go to my dataset, select a label, choose amongst some pair of hyperparameters, and train a Decision Tree model.

Persona

Existing users looking to train a model.

Feature

Be able to run a training job through our app, specifically for a decision tree model.

Business Value

Users can train a new model type from our app for datasets with categorical labels.

Tasks

  1. Script takes in input hyperparameters
  2. Script loads the dataset
  3. Script initializes a decision tree model & trains on the dataset
  4. Script saves/uploads the trained model

Acceptance Criteria

Acceptance Tests