ÎncepețiÎncepe gratuit

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

Vezi cursul

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_predictions pentru 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(____[____]))
Editează și rulează codul