mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-24 13:31:07 +00:00
30 lines
488 B
Python
30 lines
488 B
Python
|
from sklearn.neural_network import MLPClassifier
|
||
|
|
|
||
|
|
X = [[0.0, 0.0], [1.0, 1.0], [1.0, 0.0], [0.0, 1.0]]
|
||
|
|
y = [0, 1, 0, 0]
|
||
|
|
|
||
|
|
|
||
|
|
clf = MLPClassifier(
|
||
|
|
solver="lbfgs", alpha=1e-5, hidden_layer_sizes=(5, 2), random_state=1
|
||
|
|
)
|
||
|
|
|
||
|
|
clf.fit(X, y)
|
||
|
|
|
||
|
|
|
||
|
|
test = [[0.0, 0.0], [0.0, 1.0], [1.0, 1.0]]
|
||
|
|
Y = clf.predict(test)
|
||
|
|
|
||
|
|
|
||
|
|
def wrapper(Y):
|
||
|
|
"""
|
||
|
|
>>> wrapper(Y)
|
||
|
|
[0, 0, 1]
|
||
|
|
"""
|
||
|
|
return list(Y)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import doctest
|
||
|
|
|
||
|
|
doctest.testmod()
|