
46
Chapitre 1. Les fondamentaux du Machine Learning
softmax_reg = LogisticRegression(multi_class="multinomial",
solver="lbfgs", C=10)
softmax_reg.fit(X, y)
Par conséquent, la prochaine fois que vous trouverez un iris ayant des pétales de
5cm de long et de 2cm de large et que vous demanderez à votre modèle de vous dire
de quel type d’iris il s’agit, il vous répondra Iris virginica (classe 2) avec une probabilité
de 94,2% (ou Iris versicolor avec une probabilité de 5,8%):
>>> softmax_reg.predict([[5, 2]])
array([2])
>>> softmax_reg.predict_proba([[5, 2]])
array([[6.38014896e-07, 5.74929995e-02, 9.42506362e-01]])
La gure 1.28 présente les frontières ...