Olá, o meu código está dando erro de objeto não chamável. Ele fala que está na linha
confusion_matrix = confusion_matrix(test_y, model.predict(test_rfe))
Alguém saberia dizer o que está errado? Revisei algumas vezes mas não consegui encontrar o erro...
from sklearn.feature_selection import RFE
SEED = 1234
random.seed(SEED)
train_x, test_x, train_y, test_y = train_test_split(x_v6, labels, test_size=0.3)
model = RandomForestClassifier(n_estimators=100, random_state=1234)
model.fit(train_x, train_y)
rfe_selector = RFE(estimator=model, n_features_to_select=10, step=1)
rfe_selector.fit(train_x, train_y)
train_rfe = rfe_selector.transform(train_x)
test_rfe = rfe_selector.transform(test_x)
model.fit(train_rfe, train_y)
confusion_matrix = confusion_matrix(test_y, model.predict(test_rfe))
plt.figure(figsize=(10, 8))
sns.set(font_scale=1.5)
sns.heatmap(confusion_matrix, annot=True, fmt="d").set(xlabel='Predicted', ylabel='Real')
print("Our model's score is now %.2f%%" % (model.score(test_rfe, test_y) * 100))
Obrigado.