This commit is contained in:
Cindy Weng 2022-04-28 11:30:56 +01:00 коммит произвёл GitHub
Родитель a6bff2d712
Коммит e13d4845a1
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -113,7 +113,7 @@ def main():
mdl_path = Model.download(model_run, exist_ok=True)
mdl = joblib.load(os.path.join(mdl_path, 'model.pkl'))
test_accuracies[model_run.id] = mdl.score(X_test, y_test_labels)
test_accuracies[model_run.id] = mdl.score(X_test, y_test)
test_predictions[model_run.id] = [labels_dict[x] for x in mdl.predict(X_test)]
if test_accuracies: