lnallelli commited on
Commit
0c6bb49
·
verified ·
1 Parent(s): 6887450

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +18 -0
  2. requirements.txt +4 -0
  3. xgboost_model.joblib +3 -0
README.md ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # XGBoost Regression Model
2
+
3
+ This model is an XGBoost Regressor trained with GPU acceleration.
4
+
5
+ ## Model Details
6
+ - **Model Type**: XGBoost Regressor
7
+ - **Best Parameters**:
8
+ - colsample_bytree: 1
9
+ - learning_rate: 0.1
10
+ - max_depth: 3
11
+ - n_estimators: 300
12
+ - subsample: 1
13
+
14
+ ## Performance Metrics
15
+ - MAE: 637.8083
16
+ - MSE: 33597596.0758
17
+ - RMSE: 5796.3433
18
+ - R² Score: 0.8488
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ scikit-learn>=1.0.0
2
+ numpy>=1.20.0
3
+ joblib>=1.0.0
4
+ xgboost>=1.5.0
xgboost_model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:070a8d8e17a53251e18c5c91de05588cd381fc13b3e67cf010c548d69409251d
3
+ size 372408