Skip to content

yuhexiong/avocado-prices-XGBRegressor-python

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Avocado Prices XGBRegressor

Overview

  • Language: Python v3.9.15
  • Package: Scikit-Learn, xgboost
  • Model: XGBRegressor
  • Loss Function: Mean Squared Error
  • Use GridSearchCV For Automatic Parameter Tuning, cv = 5
  • Label Encoding to deal with string type column

Data Distribution

image

Result

  • Best Params: {'gamma': 0, 'max_depth': 7, 'n_estimators': 1000, 'reg_lambda': 10}
  • Mean Squared Error: 0.03500582893385328
  • R2: 0.79

Releases

No releases published

Packages

No packages published

Languages