Vasistareddy/text_classification_ktrain

Streamlining Text Classification Model with Ktrain and MLflow

text_classification_ktrain_mlflow

python distilbertTraining.py

This command in your command line creates an ml runs folder and creates an experiment id, artifacts, metrics folders, and data in them.

python bestmodelFind.py

This command finds the best model based on the metrics.