From 7f4b8b1c490571d1b2acd40c9d851f53ca5d8fa4 Mon Sep 17 00:00:00 2001 From: SteliosGian Date: Mon, 22 Nov 2021 17:55:28 +0200 Subject: [PATCH] Add best hyperparams to config file --- config.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/config.yaml b/config.yaml index f0e9338d..a911e895 100644 --- a/config.yaml +++ b/config.yaml @@ -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: @@ -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 \ No newline at end of file