junos
/
stress_at_work_analysis
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use concat instead of append which was deprecated.

master
junos 2023-05-12 16:32:08 +02:00
parent 3091328fc5
commit 87781840d4
1 changed files with 48 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
machine_learning/classification_models.py
View File

@ -60,36 +60,60 @@ class ClassificationModels:
print("\n************************************\n") print("\n************************************\n")
print("Current model:", model_title, end="\n") print("Current model:", model_title, end="\n")
print("Acc:", model["metrics"][0] / n_clusters) print("Acc:", model["metrics"][0] / n_clusters)
scores_df.append( scores_df = pd.concat(
[
scores_df,
pd.DataFrame(
{ {
"method": model_title, "method": model_title,
"metric": "test_accuracy", "metric": "test_accuracy",
"mean": model["metrics"][0] / n_clusters, "mean": model["metrics"][0] / n_clusters,
} }
),
],
ignore_index=True,
) )
print("Precision:", model["metrics"][1] / n_clusters) print("Precision:", model["metrics"][1] / n_clusters)
scores_df.append( scores_df = pd.concat(
[
scores_df,
pd.DataFrame(
{ {
"method": model_title, "method": model_title,
"metric": "test_precision", "metric": "test_precision",
"mean": model["metrics"][1] / n_clusters, "mean": model["metrics"][1] / n_clusters,
} }
),
],
ignore_index=True,
) )
print("Recall:", model["metrics"][2] / n_clusters) print("Recall:", model["metrics"][2] / n_clusters)
scores_df.append( scores_df = pd.concat(
[
scores_df,
pd.DataFrame(
{ {
"method": model_title, "method": model_title,
"metric": "test_recall", "metric": "test_recall",
"mean": model["metrics"][2] / n_clusters, "mean": model["metrics"][2] / n_clusters,
} }
),
],
ignore_index=True,
) )
print("F1:", model["metrics"][3] / n_clusters) print("F1:", model["metrics"][3] / n_clusters)
scores_df.append( scores_df = pd.concat(
[
scores_df,
pd.DataFrame(
{ {
"method": model_title, "method": model_title,
"metric": "test_f1", "metric": "test_f1",
"mean": model["metrics"][3] / n_clusters, "mean": model["metrics"][3] / n_clusters,
} }
),
],
ignore_index=True,
) )
scores = pd.concat([scores, scores_df]) scores = pd.concat([scores, scores_df])
return scores return scores