Как я могу распечатать roc_auc_score в терминале, используя ноутбук Юпитер? - PullRequest
0 голосов
/ 06 марта 2020

Здравствуйте, я использую python с блокнотом Юпитера. Когда я запускаю тренировку со Случайным лесом, я хочу напечатать sklearn.metrics roc_auc_score в терминале, это не работает, и я не понимаю почему, я комментирую разные способы достижения результата, но все работает! Я действительно ценю, если кто-то может мне помочь. Спасибо!

elif clf == 'rndforest':

# No validation data in rnd forest
x_train = np.concatenate((x_train, x_valid))
y_train = np.concatenate((y_train, y_valid))

model = RandomForestClassifier(n_estimators=int(clf_params['n_estimators']),
                               max_features=clf_params['max_features'])
model.fit(pipe_vect.transform(x_train), y_train)

datetoday = datetime.today().strftime('%d-%b-%Y-%H_%M')
model_name_save = abspath(os.path.join("models", dataset,  name_file + '-' + vect + reduction + '-rndforest'\
                               + datetoday + '.pickle'))
print("Model d'enregistrement : ", model_name_save)
#auc_score = roc_auc_score(x_train, y_valid)
#print('Performance du modèle :', auc_score)
#prediction = model.predict_proba(x_train)       
#print(roc_auc_score(y_train, prediction))
#print('ROC AUC score :', roc_auc_score(y_test, preds_evaluate[:, 1]))
#print([tolist(t) for t in x_test])
#prediction = model.predict_proba(x_test)       
#print(roc_auc_score(y_test, prediction))


if not(args.modelname is None):
    print("test1",args.modelname)
    with open(args.modelname, 'wb') as f:
        cloudpickle.dump(model, f)
else:
    print("test",model_name_save)
    with open(model_name_save, 'wb') as f:
        cloudpickle.dump(model, f)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...