Homework for RS School Machine Learning course.
This project uses Forest Cover Type dataset.
This package allows you to train model for predicting the type of cover forest.
- Clone this repository to your machine.
- Download Forest Cover Type dataset, save csv locally (default path is data/train.csv in repository's root).
- Make sure Python 3.9 and Poetry are installed on your machine.
- Install the project dependencies (run this and following commands in a terminal, from the root of a cloned repository):
poetry install --no-dev
- You can make EDA report:
poetry run eda_report -d <path to csv with data> -s <path to save report>
- Run train with the following command:
poetry run train -d <path to csv with data> -s <path to save trained model>
You can configure additional options (such as hyperparameters) in the CLI. To get a full list of them, use help:
poetry run train --help
- You can make some experements:
poetry run eval_perform
- Run MLflow UI to see the information about experiments you conducted:
poetry run mlflow ui