| tags: | |
| - autotrain | |
| - tabular | |
| - regression | |
| - tabular-regression | |
| datasets: | |
| - nicoler229/autotrain-data-renp-vcyx-5hff | |
| # Model Trained Using AutoTrain | |
| - Problem type: Tabular regression | |
| ## Validation Metrics | |
| - r2: 0.8987710422047952 | |
| - mse: 15.386801584871137 | |
| - mae: 3.1008129119873047 | |
| - rmse: 3.9226013798079378 | |
| - rmsle: 0.049014949862444 | |
| - loss: 3.9226013798079378 | |
| ## Best Params | |
| - learning_rate: 0.09858308825036341 | |
| - reg_lambda: 1.7244892825164977e-06 | |
| - reg_alpha: 0.004880162297132929 | |
| - subsample: 0.5918267532876357 | |
| - colsample_bytree: 0.6228647593929555 | |
| - max_depth: 8 | |
| - early_stopping_rounds: 440 | |
| - n_estimators: 7000 | |
| - eval_metric: rmse | |
| ## Usage | |
| ```python | |
| import json | |
| import joblib | |
| import pandas as pd | |
| model = joblib.load('model.joblib') | |
| config = json.load(open('config.json')) | |
| features = config['features'] | |
| # data = pd.read_csv("data.csv") | |
| data = data[features] | |
| predictions = model.predict(data) # or model.predict_proba(data) | |
| # predictions can be converted to original labels using label_encoders.pkl | |
| ``` | |