Prever probabilidades de críticas de filmes
Neste exercício, você vai construir uma regressão logística usando o conjunto de dados movies. Os rótulos estão armazenados no array y e as variáveis em X.
Treine o modelo nos dados de treino. Em vez de prever classes, preveja as probabilidades de que cada instância no conjunto de teste pertença a cada uma das duas classes.
As funções de regressão logística e de divisão treino/teste já foram importadas para você.
Este exercício faz parte do curso
Análise de Sentimentos em Python
Instruções do exercício
- Divida os dados em conjuntos de treino e teste.
- Treine um modelo de regressão logística.
- Preveja as probabilidades para a classe 0 e para a classe 1 nos dados de teste. A classe 0 fica na primeira coluna das probabilidades previstas, e a classe 1 na segunda.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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])