Prezice probabilitățile recenziilor de filme
În acest exercițiu, vei construi un model de regresie logistică folosind setul de date movies. Etichetele sunt stocate în array-ul y, iar caracteristicile în X.
Antrenează modelul pe datele de antrenament. În loc să prezici clase, prezice probabilitățile ca fiecare instanță din setul de testare să aparțină uneia dintre cele două clase.
Funcțiile pentru regresia logistică și pentru împărțirea în seturi de antrenament/testare au fost deja importate.
Acest exercițiu face parte din cursul
Analiza sentimentelor în Python
Instrucțiuni pentru exercițiu
- Împarte datele în set de antrenament și set de testare.
- Antrenează un model de regresie logistică.
- Prezice probabilitățile pentru clasa 0 și pentru clasa 1 pe datele de testare. Clasa 0 se află în prima coloană a probabilităților prezise, iar clasa 1 în a doua.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# Split into training and testing
X_train, X_test, y_train, y_test = ____(___, ___, test_size=0.2, random_state=321)
# Train a logistic regression
log_reg = ____.____
# Predict the probability of the 0 class
prob_0 = log_reg.____[:, ____]
# Predict the probability of the 1 class
prob_1 = log_reg.____[:, ____]
print("First 10 predicted probabilities of class 0: ", prob_0[:10])
print("First 10 predicted probabilities of class 1: ", prob_1[:10])