Spaces:
Sleeping
Sleeping
Update README.md
Browse files
README.md
CHANGED
@@ -86,11 +86,11 @@ All the attributes are encoded and normalized before splitting into train and te
|
|
86 |
|
87 |
**Milestone 2:**
|
88 |
|
89 |
-
For milestone 2, I
|
90 |
|
91 |
**Milestone 3:**
|
92 |
|
93 |
-
|
94 |
|
95 |
**References:**
|
96 |
|
|
|
86 |
|
87 |
**Milestone 2:**
|
88 |
|
89 |
+
For milestone 2, I used an XGBoost Model with objective="reg:squarederror" and max_depth=3. The RMSE score is 28986.
|
90 |
|
91 |
**Milestone 3:**
|
92 |
|
93 |
+
For milestone 3, I used light gradient boosting machine (LGBM) with default parameters for baseline and hyperparameter-tuned with Optuna for the optimized model. The results are stated at the beginning of my readme file.
|
94 |
|
95 |
**References:**
|
96 |
|