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