update app.py
Browse files
app.py
CHANGED
|
@@ -16,6 +16,26 @@ client.set_token("zE9NjEOgUMzH9K7mjiGBaPJiNwJLjSM53LevarRK", "token")
|
|
| 16 |
dataset_dir = "dataset_audio"
|
| 17 |
os.makedirs(dataset_dir, exist_ok=True)
|
| 18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
# -------- Charger les objets sauvegardés --------
|
| 20 |
# Music
|
| 21 |
scaler_samplerate_music = joblib.load("music/scaler_music_samplerate.joblib")
|
|
@@ -40,23 +60,6 @@ glove_model = api.load("glove-wiki-gigaword-100")
|
|
| 40 |
|
| 41 |
# -------- Fonctions --------
|
| 42 |
|
| 43 |
-
class AvgRatingTransformer:
|
| 44 |
-
def __init__(self, est, class_mapping=None):
|
| 45 |
-
self.est = est
|
| 46 |
-
if class_mapping is None:
|
| 47 |
-
self.class_mapping = {0:"MissedInfo", 1:"Low", 2:"Medium", 3:"High"}
|
| 48 |
-
else:
|
| 49 |
-
self.class_mapping = class_mapping
|
| 50 |
-
|
| 51 |
-
def transform(self, X):
|
| 52 |
-
X = X.copy()
|
| 53 |
-
mask_non_zero = X != 0
|
| 54 |
-
Xt = np.zeros_like(X, dtype=int)
|
| 55 |
-
if mask_non_zero.any():
|
| 56 |
-
Xt[mask_non_zero] = self.est.transform(X[mask_non_zero].reshape(-1,1)).flatten() + 1
|
| 57 |
-
# Appliquer le mapping
|
| 58 |
-
X_transformed = np.array([self.class_mapping.get(v, "MissedInfo") for v in Xt])
|
| 59 |
-
return X_transformed
|
| 60 |
|
| 61 |
|
| 62 |
def fetch_sound_metadata(sound_url):
|
|
|
|
| 16 |
dataset_dir = "dataset_audio"
|
| 17 |
os.makedirs(dataset_dir, exist_ok=True)
|
| 18 |
|
| 19 |
+
class AvgRatingTransformer:
|
| 20 |
+
def __init__(self, est, class_mapping=None):
|
| 21 |
+
self.est = est
|
| 22 |
+
if class_mapping is None:
|
| 23 |
+
self.class_mapping = {0:"MissedInfo", 1:"Low", 2:"Medium", 3:"High"}
|
| 24 |
+
else:
|
| 25 |
+
self.class_mapping = class_mapping
|
| 26 |
+
|
| 27 |
+
def transform(self, X):
|
| 28 |
+
X = X.copy()
|
| 29 |
+
mask_non_zero = X != 0
|
| 30 |
+
Xt = np.zeros_like(X, dtype=int)
|
| 31 |
+
if mask_non_zero.any():
|
| 32 |
+
Xt[mask_non_zero] = self.est.transform(X[mask_non_zero].reshape(-1,1)).flatten() + 1
|
| 33 |
+
# Appliquer le mapping
|
| 34 |
+
X_transformed = np.array([self.class_mapping.get(v, "MissedInfo") for v in Xt])
|
| 35 |
+
return X_transformed
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
|
| 39 |
# -------- Charger les objets sauvegardés --------
|
| 40 |
# Music
|
| 41 |
scaler_samplerate_music = joblib.load("music/scaler_music_samplerate.joblib")
|
|
|
|
| 60 |
|
| 61 |
# -------- Fonctions --------
|
| 62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
|
| 64 |
|
| 65 |
def fetch_sound_metadata(sound_url):
|