import pandas as pd from sklearn.neural_network import MLPClassifier data = pd.read_csv("data.csv", sep=",") clf = MLPClassifier(solver="sgd",random_state=0,max_iter=10000) clf.fit(data[['x1', 'x2']], data['x3']) pred = clf.predict(data[['x1', 'x2']]) print (pred)