ソースを参照

Enable merge from rfe script

Jérôme BUISINE 2 年 前
コミット
bc9d7037b7
2 ファイル変更1 行追加1 行削除
  1. 1 0
      check_random_forest_perfomance_rfe.py
  2. 0 1
      train_model_attributes.py

+ 1 - 0
check_random_forest_perfomance_rfe.py

@@ -98,6 +98,7 @@ def train_predict_selector(model, x_train, y_train, x_test, y_test):
     start = datetime.datetime.now()
 
     print("Using Select from model with Random Forest")
+
     selector = RFECV(estimator=model, min_features_to_select=13, verbose=1, n_jobs=-1)
     selector.fit(x_train, y_train)
     x_train_transformed = selector.transform(x_train)

+ 0 - 1
train_model_attributes.py

@@ -114,7 +114,6 @@ def main():
 
     print()
 
-    return
     #######################
     # 2. Construction of the model : Ensemble model structure
     #######################