导入相关库:
from sklearn.neural_network import MLPClassifier
设置训练样本以及对应的标注:
>>> X = [[0., 0.], [1., 1.]] >>> y = [0, 1]
实例化多层神经网络算法,其中隐藏层一共两层,对应神经元个数分别为5个和2个:
>>> clf = MLPClassifier(solver='lbfgs', alpha=1e-5, ... hidden_layer_sizes=(5, 2), random_state=1) >>> clf.fit(X, y)
预测结果:
>>> clf.predict([[2., 2.], [-1., -2.]]) array([1, 0])