FelixPhilip commited on
Commit
e574555
·
1 Parent(s): 607473c
Files changed (1) hide show
  1. Oracle/deepfundingoracle.py +3 -1
Oracle/deepfundingoracle.py CHANGED
@@ -512,7 +512,9 @@ def train_predict_weight(df):
512
  # Split data into train/test sets
513
  X_train, X_test, y_train_log, y_test_log = train_test_split(X, y_log, test_size=0.2, random_state=42)
514
 
515
- pipeline = Pipeline(["rf", RandomForestRegressor(random_state=42)])
 
 
516
  # Hyperparameter tuning using GridSearchCV
517
  param_dist = {
518
  "rf__n_estimators": [100, 300, 500, 800, 1000],
 
512
  # Split data into train/test sets
513
  X_train, X_test, y_train_log, y_test_log = train_test_split(X, y_log, test_size=0.2, random_state=42)
514
 
515
+ pipeline = Pipeline([
516
+ ("rf", RandomForestRegressor(random_state=42))
517
+ ])
518
  # Hyperparameter tuning using GridSearchCV
519
  param_dist = {
520
  "rf__n_estimators": [100, 300, 500, 800, 1000],