如何在 Google Colab 中缓存 GridSearchCV 优化器结果?

数据挖掘 网格搜索 合作实验室
2021-10-07 14:19:10

GridSearchCV()每次重新加载页面时我都必须执行单元格,这需要很长时间。是否有存储 GridSearchCV 结果的缓存机制,以便我可以在重新加载时不执行单元格的情况下使用它?

代码:

 neigh_clf = KNeighborsClassifier()
 grid_search = GridSearchCV(neigh_clf, param_grid, cv=5,verbose=3,n_jobs=-1)
 grid_search.fit(X_train, y_train)
 grid_search.best_params_

我正在尝试在此处保存 grid_search 对象,以便在重新加载后可以检索 best_params_ 值

1个回答

我们可以通过 Google Colaboratory 保存训练好的模型或任何其他文件。

我如何使用它?

我已经用 Google Colaboratory 笔记本映射了我的 Google Drive,并将经过训练的模型保存为其中的 pickle 文件。您可以创建一个文件并将数据保存在其中。

如何将 Google Drive 与 Google Colaboratory notebook 集成?

#Add and execute below mentioned line of code in Google colaboratory notebook cell.
from google.colab import drive
drive.mount('ndrive')
#Once you execute these two lines, it will ask you to authorize it. Please follow the steps as visible on the Google Colaboratory Notebook.

而已 :)。

几个重要的点:

  1. 现在您可以运行通常的 linux 命令来浏览目录。
  2. 默认情况下,它映射到内容文件夹,您可以在其中创建子目录。
  3. 您也可以通过 Google-Drive 应用程序访问这些子目录。