Answers for "apply function to output of scikit learn model"

predict method sklearn

from sklearn import neighbors, datasets
iris = datasets.load_iris()
X, y = iris.data, iris.target
knn = neighbors.KNeighborsClassifier(n_neighbors=1)
knn.fit(X, y)
# What kind of iris has 3cm x 5cm sepal and 4cm x 2cm petal?
print(iris.target_names[knn.predict([[3, 5, 4, 2]])])

sklearn - check the name of a machine learning

# Define the model
model1 = LogisticRegression()

# Check the name of the model
# It is useful when we are dealing with many alternative models
model1.__class__.__name__