@@ -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)
@@ -114,7 +114,6 @@ def main():
print()
- return
#######################
# 2. Construction of the model : Ensemble model structure