Skip to content

Commit

Permalink
Add best hyperparams to config file
Browse files Browse the repository at this point in the history
  • Loading branch information
SteliosGian committed Nov 22, 2021
1 parent 50892da commit 7f4b8b1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ modeling:
stratify_by: "neighbourhood_group"
# Maximum number of features to consider for the TFIDF applied to the title of the
# insertion (the column called "name")
max_tfidf_features: 5
max_tfidf_features: 10
# NOTE: you can put here any parameter that is accepted by the constructor of
# RandomForestRegressor. This is a subsample, but more could be added:
random_forest:
Expand All @@ -33,6 +33,6 @@ modeling:
# Here -1 means all available cores
n_jobs: -1
criterion: mae
max_features: 0.5
max_features: 0.33
# DO not change the following
oob_score: true

0 comments on commit 7f4b8b1

Please sign in to comment.