14.2 示例:hello world!神经网络

导入相关库:


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])