python-3.x - 卡在 jupyter 笔记本中

Question

我是机器学习的新手，我开始学习本教程来了解员工流失。我按照这些步骤操作，但我的代码卡在 20 以下的单元格上。我唯一更改的是删除 idd 参数，但它适用于它上面的那个，而不是这个。

代码图像

rf_classifier = RandomForestClassifier(class_weight = "balanced",
                                       random_state=7)
param_grid = {'n_estimators': [50, 75, 100, 125, 150, 175],
              'min_samples_split':[2,4,6,8,10],
              'min_samples_leaf': [1, 2, 3, 4],
              'max_depth': [5, 10, 15, 20, 25]}

grid_obj = GridSearchCV(rf_classifier,
                        return_train_score=True,
                        param_grid=param_grid,
                        scoring='roc_auc',
                        cv=10)

grid_fit = grid_obj.fit(X_train, y_train)
rf_opt = grid_fit.best_estimator_

print('='*20)
print("best params: " + str(grid_obj.best_estimator_))
print("best params: " + str(grid_obj.best_params_))
print('best score:', grid_obj.best_score_)
print('='*20)

我正在关注的项目的 Github 链接。

标签： python-3.xjupyter-notebookclassificationrandom-forest