반응형
2021/01/25
-
MLP에서 적절한 hidden unit 개수 산정하기프로그래밍/기계학습 2021. 1. 25. 20:23
sklearn MLP 알고리즘에서 적절한 hidden unit 개수 산정하기 skearn에서 MLP classifier나 regressor를 사용할때 hiddenunit 개수를 몇 개로 시작해야 해야하는지에 대해서 좋은 대답이 있어서 기록해둔다. model_mlp = MLPClassifier(solver='lbfgs', alpha=1e-5, hidden_layer_sizes=(hiddenunit, 5), random_state=1) If the NN is a regressor, then the output layer has a single node. If the NN is a classifier, then it also has a single node unless softmax is used in whi..