Spaces:
Running
Running
Commit
·
e574555
1
Parent(s):
607473c
Oracle
Browse files
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([
|
|
|
|
|
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],
|