GitLab wurde erfolgreich aktualisiert. Durch regelmäßige Updates bleibt das THM GitLab sicher. Danke für Ihre Geduld.

Commit fc0a6aa9 authored by Jens Plüddemann's avatar Jens Plüddemann

commenting

parent a2c2173f
......@@ -124,6 +124,8 @@ class LogReg:
regressor = LogisticRegression().fit(X_train_scaled, y_train)
# Errechnen der roc curve
# [:, -1] nimmt aus einer Liste an Listen immer den letzten Wert
# So wird hier aus [[1, 2], [3, 4]] => [2, 4]
logit_roc_auc = roc_auc_score(y_test, regressor.predict(X_test_scaled))
fpr, tpr, threshholds = roc_curve(y_test, regressor.predict_proba(X_test_scaled)[:, -1])
......@@ -145,7 +147,7 @@ class LogReg:
if __name__ == '__main__':
data = LogReg('../../res/Data_LogReg.csv')
# data.plot_initial()
data.plot_initial()
# data.plot_logistic_regression()
# data.plot_confusion_matrix()
data.plot_roc_curve()
\ No newline at end of file
# data.plot_roc_curve()
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