brendabor commited on
Commit
981503f
·
1 Parent(s): 18afb5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -57,8 +57,12 @@ def hybrid_recommendation(song_index):
57
  features_for_similarity = df[['danceability', 'energy', 'key', 'loudness', 'mode', 'speechiness',
58
  'acousticness', 'instrumentalness', 'liveness', 'valence', 'tempo',
59
  'duration_ms', 'time_signature']].values
 
 
 
 
60
 
61
- cosine_similarities = cosine_similarity([emotion_category], features_for_similarity).flatten()
62
 
63
  # Combine recommendations from both models
64
  combined_indices = np.argsort(-np.concatenate([knn_recommendations, cosine_similarities]))
 
57
  features_for_similarity = df[['danceability', 'energy', 'key', 'loudness', 'mode', 'speechiness',
58
  'acousticness', 'instrumentalness', 'liveness', 'valence', 'tempo',
59
  'duration_ms', 'time_signature']].values
60
+ scaler = StandardScaler()
61
+ audio_features_scaled = scaler.fit_transform(features_for_similarity)
62
+ # Combine mood and audio features
63
+ combined_features = pd.concat([emotion_category, pd.DataFrame(audio_features_scaled)], axis=1)
64
 
65
+ cosine_similarities = cosine_similarity(combined_features)
66
 
67
  # Combine recommendations from both models
68
  combined_indices = np.argsort(-np.concatenate([knn_recommendations, cosine_similarities]))