python - 如何在 Scikit 上训练时获取权重矩阵

标签 python machine-learning scikit-learn neural-network

我正在使用 Scikit 训练 MLPClassifier。假设我想在 MNIST 上训练 5 个 epoch,其中一个隐藏层有 100 个神经元。

如果我执行“mlp = MLPClassifier(...)”,然后执行“mlp.fit(train,test)”,那么我可以使用“mlp.coefs_”获得训练后的权重。

但我想要的是训练期间每个时期后获得的权重矩阵序列。因此,如果我训练 5 个 epoch,我会想要一个大小为 5 的列表,其中包含权重矩阵的历史记录。

这可以用 scikit 实现吗?或者我应该使用 Keras?

最佳答案

一种选择是使用您想要执行的一小部分 epoch 来训练您的模型。

存储参数。

然后继续使用 Warm_start = True 参数训练您的模型。您可以这样做,直到获得所需的纪元总数。

在 sci-kit learns 实现的上下文中,max_iter 参数将是纪元。此链接中引用了此内容。
https://stats.stackexchange.com/questions/284491/are-the-epochs-equivalent-to-the-iterations

关于python - 如何在 Scikit 上训练时获取权重矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59904720/

相关文章:

python - 使用 splinter python 保存图像元素

Python3 exchangelib 连接错误 : _ssl. c :749?

python - 对矩阵使用降维

python - scikit : Wrong prediction for this case

python - 获取每个文档的最高术语 - scikit tf-idf

python - 如何在 TensorFlow 中将信号与一维内核进行卷积?

machine-learning - matlab中的最小二乘SVM

python - 如何使用 scikit-learn silhouette score 计算 scipy 的 fcluster 的 Silhouette Score?

python - 并行作业不会在 scikit-learn 的 GridSearchCV 中完成

python scipy lesssq 适合复数