Downloads · 30 days
0
mmobi/Forest_Regressor_v1
Forest_Regressor_v1 is a tabular regression model from mmobi. Use it for the tabular regression task on the model card, and read the license before you ship it in a product. It is set up for sklearn. The card lists the license as mit.
Модель машинного обучения для прогнозирования популярности музыкальных треков на основе акустических характеристик из Spotify API.
Downloads · 30 days
0
Access
Public
Updated Dec 7, 2025
Repo size
242 MB
Likes
0
Public
Click a slice to open those files.
.pkl242 MB · 100%
From the Hugging Face model README
Модель машинного обучения для прогнозирования популярности музыкальных треков на основе акустических характеристик из Spotify API.
Random Forest Regressor обученная на 114,000 треков для предсказания популярности (0-100).
n_estimators: 50max_depth: 30min_samples_split: 5min_samples_leaf: 2max_features: 'sqrt'| Метрика | Train | Test | Оценка |
|---|---|---|---|
| MAE | 6.40 | 11.44 | ✅ Хорошо |
| R² | - | 0.52 | ⚠️ Средне |
| Стабильность CV | 0.98% | - | ✅ Отлично |
Погрешность: ±11 пунктов популярности в среднем
import pickle
import pandas as pd
# Загрузка модели
with open('final_random_forest_model.pkl', 'rb') as f:
model = pickle.load(f)
# Подготовка данных (пример)
features = pd.DataFrame({
'danceability': [0.7],
'energy': [0.8],
'loudness': [-5.0],
'speechiness': [0.05],
'acousticness': [0.1],
'instrumentalness': [0.0],
'liveness': [0.1],
'valence': [0.6],
'tempo': [120.0],
'duration_ms': [200000],
'key': [5],
'mode': [1],
'time_signature': [4]
# + производные признаки (см. feature engineering)
})
# Предсказание
popularity = model.predict(features)
print(f"Предсказанная популярность: {popularity[0]:.1f}/100")
Предсказанная популярность:
0-30 → ❌ Слабый потенциал
30-50 → ⚠️ Средний потенциал
50-70 → ✅ Хороший потенциал
70+ → 🎯 Высокий потенциал (модель редко так предсказывает)
reports/model_analysis.md в репозиторииTECHNICAL_SPECIFICATION.mdMIT License
mmobik