shravanda/superkart-sales-data
Viewer • Updated • 26.3k • 60
How to use shravanda/superkart-sales-model with Scikit-learn:
from huggingface_hub import hf_hub_download
import joblib
model = joblib.load(
hf_hub_download("shravanda/superkart-sales-model", "sklearn_model.joblib")
)
# only load pickle files from sources you trust
# read more about it here https://skops.readthedocs.io/en/stable/persistence.htmlPredicts Product_Store_Sales_Total (revenue of a product in a store) for SuperKart.
Serialized scikit-learn Pipeline (preprocessing + RandomForest) tuned with 5-fold GridSearchCV.
| Metric (test set) | Value |
|---|---|
| RMSE | 280.45 |
| MAE | 105.72 |
| R2 | 0.9311 |
| MAPE | 0.0384 |
Best hyper-parameters: {'max_depth': None, 'max_features': 1.0, 'min_samples_leaf': 4, 'n_estimators': 100}
import joblib
from huggingface_hub import hf_hub_download
model = joblib.load(hf_hub_download("shravanda/superkart-sales-model", "best_superkart_model_v1.joblib"))