/genre_classification

The Machine Learning pipeline

Primary LanguagePythonMIT LicenseMIT

Genre Classification

The Machine Learning pipeline with MLflow 🤖

Run

Given you have navigated into the root folder of this project. This will run the complete pipeline.

mlflow run .

Override pipeline steps and only run random_forest step.

mlflow run . -P hydra_options="main.execute_steps='random_forest'"

Run from GitHub

If you are logged into Weight & Biases you can run this MLflow pipeline from this GitHub repository.

mlflow run -v 1.0.0 https://github.com/marcusholmgren/genre_classification