Predicții de clasificare
În validarea modelelor, este adesea important să știi mai multe despre predicții decât simpla clasificare finală. Când prezici câștigătorul unui meci, majoritatea oamenilor sunt interesați și de cât de probabilă este victoria unei echipe.
| Probabilitate | Predicție | Semnificație |
|---|---|---|
| 0 < .50 | 0 | Echipa pierde |
| .50 + | 1 | Echipa câștigă |
În acest exercițiu, explorezi metodele .predict() și .predict_proba() folosind setul de date tic_tac_toe. Prima metodă returnează predicția privind câștigătorul jocului (Jucătorul Unu), iar a doua metodă oferă probabilitatea ca Jucătorul Unu să câștige. Folosește rfc ca model de clasificare prin păduri aleatoare.
Acest exercițiu face parte din cursul
Validarea modelelor în Python
Instrucțiuni pentru exercițiu
- Creează două array-uri de predicții: unul pentru valorile de clasificare și unul pentru probabilitățile estimate.
- Folosește metoda
.value_counts()pentru o serie pandas, pentru a afișa numărul de observații atribuite fiecărei clase. - Afișează prima observație din
probability_predictionspentru a vedea cum sunt structurate probabilitățile.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Fit the rfc model.
rfc.fit(X_train, y_train)
# Create arrays of predictions
classification_predictions = rfc.____(X_test)
probability_predictions = rfc.____(X_test)
# Print out count of binary predictions
print(pd.Series(____).____())
# Print the first value from probability_predictions
print('The first predicted probabilities are: {}'.format(____[____]))