python-scikit-learnLogistic regression example

Usage example

from sklearn import datasets, linear_model, model_selection

X, y = datasets.load_iris(return_X_y=True)
X_train, X_test, y_train, y_test = model_selection.train_test_split(X, y)

model = linear_model.LogisticRegression()
model.fit(X_train, y_train)
y_pred = model.predict(X_test)

print(y_pred)

output

[1 2 1 0 2 1 2 2 0 2 1 0 0 2 0 1 0 1 1 2 0 0 0 1 1 0 0 1 1 1 1 2 0 2 0 1 2
 0]