【发布时间】:2020-03-22 04:34:10
【问题描述】:
请看下面的例子:
parameter_space = {
'hidden_layer_sizes': [(200,100)],
'activation': ['tanh', 'relu'],
'solver': ['sgd', 'adam'],
'alpha': [0.0001, 0.05],
'learning_rate': ['constant','adaptive'],
}
clf = GridSearchCV(mlp, parameter_space, n_jobs=-1, cv=3)
问题是要了解我的 ANN 是如何留在内部的。
我希望这个示例使用 20 个神经元作为输入层,两个隐藏层分别为 200 和 100 个神经元,以及一个带有 1 个神经元的输出层,因为我的问题是二元分类。
输入层 (20) --- 隐藏层 1 (200) --- 隐藏层 2 (100) --- 输出层 (1)
我知道在“hidden_layer_sizes”中只会插入隐藏层,但是我们如何指示输入和输出层呢?以及“GridSearchCV”将如何交替隐藏层中的神经元数量以找到最佳配置?
【问题讨论】:
标签: python scikit-learn gridsearchcv mlp