שינוי שם מודל זיהוי סוג

NHLOCAL · Sep 3, 2024 · 3d36266 · 3d36266
1 parent af4a4b9
commit 3d36266
Show file tree

Hide file tree

Showing 16 changed files with 1 addition and 1 deletion.
diff --git a/...ssification/model_creation/build_model.py → ..._entity_clf/model_creation/build_model.py b/...ssification/model_creation/build_model.py → ..._entity_clf/model_creation/build_model.py
diff --git a/...classification/model_creation/dataset.csv → music_entity_clf/model_creation/dataset.csv b/...classification/model_creation/dataset.csv → music_entity_clf/model_creation/dataset.csv
diff --git a/...ation/model_creation/music_classifier.pkl → ...y_clf/model_creation/music_entity_clf.pkl b/...ation/model_creation/music_classifier.pkl → ...y_clf/model_creation/music_entity_clf.pkl
diff --git a/...tion/model_running/classified_strings.csv → ..._clf/model_running/classified_strings.csv b/...tion/model_running/classified_strings.csv → ..._clf/model_running/classified_strings.csv
diff --git a/...lassification/model_running/test_model.py → music_entity_clf/model_running/test_model.py b/...lassification/model_running/test_model.py → music_entity_clf/model_running/test_model.py
diff --git a/...classification/model_running/test_set.csv → music_entity_clf/model_running/test_set.csv b/...classification/model_running/test_set.csv → music_entity_clf/model_running/test_set.csv
diff --git a/...classification/model_running/try_model.py → music_entity_clf/model_running/try_model.py b/...classification/model_running/try_model.py → music_entity_clf/model_running/try_model.py
diff --git a/...raw_data_creation/data/raw_album_data.txt → ...raw_data_creation/data/raw_album_data.txt b/...raw_data_creation/data/raw_album_data.txt → ...raw_data_creation/data/raw_album_data.txt
diff --git a/...aw_data_creation/data/raw_album_data2.txt → ...aw_data_creation/data/raw_album_data2.txt b/...aw_data_creation/data/raw_album_data2.txt → ...aw_data_creation/data/raw_album_data2.txt
diff --git a/...ta_creation/data/raw_random_text_data.txt → ...ta_creation/data/raw_random_text_data.txt b/...ta_creation/data/raw_random_text_data.txt → ...ta_creation/data/raw_random_text_data.txt
diff --git a/.../raw_data_creation/data/raw_song_data.txt → .../raw_data_creation/data/raw_song_data.txt b/.../raw_data_creation/data/raw_song_data.txt → .../raw_data_creation/data/raw_song_data.txt
diff --git a/...a_creation/scripts/generate_album_data.py → ...a_creation/scripts/generate_album_data.py b/...a_creation/scripts/generate_album_data.py → ...a_creation/scripts/generate_album_data.py
diff --git a/...tion/scripts/generate_random_text_data.py → ...tion/scripts/generate_random_text_data.py b/...tion/scripts/generate_random_text_data.py → ...tion/scripts/generate_random_text_data.py
diff --git a/...ta_creation/scripts/generate_song_data.py → ...ta_creation/scripts/generate_song_data.py b/...ta_creation/scripts/generate_song_data.py → ...ta_creation/scripts/generate_song_data.py
diff --git a/src/core/ai_models.py b/src/core/ai_models.py
@@ -35,7 +35,7 @@ def load_models(self):
 
         # Load the sklearn model
         try:
-            model_path = 'models/music_classifier.pkl'
+            model_path = 'models/music_entity_clf/music_entity_clf.pkl'
             with open(model_path, 'rb') as model_file:
                 self.sklearn_model = pickle.load(model_file)
             self.logger.debug("Loaded sklearn model successfully")

diff --git a/src/core/models/music_classifier.pkl → ...els/music_entity_clf/music_entity_clf.pkl b/src/core/models/music_classifier.pkl → ...els/music_entity_clf/music_entity_clf.pkl