Spaces:
Build error
Build error
Convert list back to array
Browse files- vendiscore.py +3 -3
vendiscore.py
CHANGED
@@ -145,11 +145,11 @@ class VendiScore(evaluate.Metric):
|
|
145 |
device="cpu",
|
146 |
):
|
147 |
if score_K:
|
148 |
-
vs = vendi.score_K(samples, normalize=normalize)
|
149 |
elif score_dual:
|
150 |
-
vs = vendi.score_dual(samples, normalize=normalize)
|
151 |
elif score_X:
|
152 |
-
vs = vendi.score_X(samples, normalize=normalize)
|
153 |
elif type(k) == str and k == "ngram_overlap":
|
154 |
vs = text_utils.ngram_vendi_score(
|
155 |
samples, ns=ns, tokenizer=tokenizer
|
|
|
145 |
device="cpu",
|
146 |
):
|
147 |
if score_K:
|
148 |
+
vs = vendi.score_K(np.array(samples), normalize=normalize)
|
149 |
elif score_dual:
|
150 |
+
vs = vendi.score_dual(np.array(samples), normalize=normalize)
|
151 |
elif score_X:
|
152 |
+
vs = vendi.score_X(np.array(samples), normalize=normalize)
|
153 |
elif type(k) == str and k == "ngram_overlap":
|
154 |
vs = text_utils.ngram_vendi_score(
|
155 |
samples, ns=ns, tokenizer=tokenizer
|