COSC481W-2024Fall / MLite

0 stars 0 forks source link

ML - Training Script - Decision Trees #8

Open Mohammad4844 opened 1 week ago

Mohammad4844 commented 1 week 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