Update modeling_apex.py
Browse files- modeling_apex.py +5 -5
modeling_apex.py
CHANGED
|
@@ -205,18 +205,18 @@ class APEXModel(PreTrainedModel):
|
|
| 205 |
for task in preds
|
| 206 |
}
|
| 207 |
|
| 208 |
-
print(f"\n{'
|
| 209 |
print(f" APEX Predictions")
|
| 210 |
-
print(f"{'
|
| 211 |
print(f"\n Popularity:")
|
| 212 |
-
print(f" {'
|
| 213 |
print(f" {'Streams Score':<20} {results['score_streams']:>8.2f} / 100")
|
| 214 |
print(f" {'Likes Score':<20} {results['score_likes']:>8.2f} / 100")
|
| 215 |
print(f"\n Aesthetic Quality:")
|
| 216 |
-
print(f" {'
|
| 217 |
for dim in ["coherence", "musicality", "memorability", "clarity", "naturalness"]:
|
| 218 |
print(f" {dim.capitalize():<20} {results[dim]:>8.2f} / 5.00")
|
| 219 |
-
|
| 220 |
|
| 221 |
if save_json:
|
| 222 |
with open(save_json, "w") as f:
|
|
|
|
| 205 |
for task in preds
|
| 206 |
}
|
| 207 |
|
| 208 |
+
print(f"\n{'─'*50}")
|
| 209 |
print(f" APEX Predictions")
|
| 210 |
+
print(f"{'─'*50}")
|
| 211 |
print(f"\n Popularity:")
|
| 212 |
+
print(f" {'-'*40}")
|
| 213 |
print(f" {'Streams Score':<20} {results['score_streams']:>8.2f} / 100")
|
| 214 |
print(f" {'Likes Score':<20} {results['score_likes']:>8.2f} / 100")
|
| 215 |
print(f"\n Aesthetic Quality:")
|
| 216 |
+
print(f" {'-'*40}")
|
| 217 |
for dim in ["coherence", "musicality", "memorability", "clarity", "naturalness"]:
|
| 218 |
print(f" {dim.capitalize():<20} {results[dim]:>8.2f} / 5.00")
|
| 219 |
+
|
| 220 |
|
| 221 |
if save_json:
|
| 222 |
with open(save_json, "w") as f:
|