Commit d7a62abf authored by Feliks Kiszkurno's avatar Feliks Kiszkurno
Browse files

I should be back in the main branch.

parent 23aae255
......@@ -78,18 +78,15 @@ def run_classification(test_training, test_prediction, test_results, clf, clf_na
print(test_name)
x_train, y_train = slopestabilityML.preprocess_data(test_results[test_name])
# Train classifier
clf_pipeline.fit(x_train, y_train)
y_pred = clf_pipeline.predict(x_train)
score_training1 = clf_pipeline.score(x_train, y_train)
score_training = accuracy_score(y_train, y_pred)
if score_training1 == score_training:
print('MATCH!')
else:
print('MISMATCH!')
accuracy_result_training.append(score_training * 100)
accuracy_labels_training.append(test_name)
clf_pipeline.fit(x_train, y_train)
y_pred = clf_pipeline.predict(x_train)
score_training1 = clf_pipeline.score(x_train, y_train)
score_training = accuracy_score(y_train, y_pred)
accuracy_result_training.append(score_training * 100)
accuracy_labels_training.append(test_name)
slopestabilityML.plot_class_overview(test_results[test_name], test_name, y_train, y_pred, clf_name, training=True)
slopestabilityML.plot_class_overview(test_results[test_name], test_name, y_train, y_pred, clf_name, training=True)
result_class = {}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment