Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ MODEL_NAMES = get_model_names(DATA)
|
|
| 37 |
# Define the six metrics in the desired order.
|
| 38 |
METRICS = ["Correctness", "Completeness", "Conciseness", "Helpfulness", "Honesty", "Harmlessness"]
|
| 39 |
|
| 40 |
-
def generate_heatmap_image(model_entry, max_size=(
|
| 41 |
"""
|
| 42 |
For a given model entry, extract the six metrics and compute a 6x6 similarity matrix
|
| 43 |
using the definition: similarity = 1 - |v_i - v_j|, then return the heatmap as a PIL image.
|
|
|
|
| 37 |
# Define the six metrics in the desired order.
|
| 38 |
METRICS = ["Correctness", "Completeness", "Conciseness", "Helpfulness", "Honesty", "Harmlessness"]
|
| 39 |
|
| 40 |
+
def generate_heatmap_image(model_entry, max_size=(20, 20)):
|
| 41 |
"""
|
| 42 |
For a given model entry, extract the six metrics and compute a 6x6 similarity matrix
|
| 43 |
using the definition: similarity = 1 - |v_i - v_j|, then return the heatmap as a PIL image.
|