Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
9e7288d
1
Parent(s): da6f7d4
Update model.py
Browse files- src/models/model.py +1 -3
src/models/model.py
CHANGED
|
@@ -75,9 +75,7 @@ class DataModule(Dataset):
|
|
| 75 |
)
|
| 76 |
|
| 77 |
labels = output_encoding["input_ids"]
|
| 78 |
-
labels[
|
| 79 |
-
labels == 0
|
| 80 |
-
] = -100
|
| 81 |
|
| 82 |
return dict(
|
| 83 |
keywords=data_row["input_text"],
|
|
|
|
| 75 |
)
|
| 76 |
|
| 77 |
labels = output_encoding["input_ids"]
|
| 78 |
+
labels[labels == 0] = -100
|
|
|
|
|
|
|
| 79 |
|
| 80 |
return dict(
|
| 81 |
keywords=data_row["input_text"],
|