fixed y_test
This commit is contained in:
Родитель
a6bff2d712
Коммит
e13d4845a1
|
@ -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:
|
||||
|
|
Загрузка…
Ссылка в новой задаче