diff --git a/lc_classifier/classifier/models.py b/lc_classifier/classifier/models.py index 5f73e68..1bab072 100644 --- a/lc_classifier/classifier/models.py +++ b/lc_classifier/classifier/models.py @@ -108,7 +108,7 @@ def load_model(self, directory: str) -> None: class HierarchicalRandomForest(BaseClassifier): - MODEL_NAME = "hierarchical_random_forest" + MODEL_NAME = "hierarchical_rf" MODEL_VERSION = "1.1.0" MODEL_VERSION_NAME = f"{MODEL_NAME}_{MODEL_VERSION}" MODEL_PICKLE_PATH = os.path.join(PICKLE_PATH, f"{MODEL_VERSION_NAME}")