NIIHAAD commited on
Commit
dfedcbf
·
1 Parent(s): 009cd45

update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -17
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):