fix error in code
Browse files
README.md
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
---
|
| 2 |
language: "en"
|
| 3 |
datasets:
|
| 4 |
-
-
|
| 5 |
tags:
|
| 6 |
- bert
|
| 7 |
- classification
|
|
@@ -32,7 +32,7 @@ for i, s in enumerate(desc_sentences):
|
|
| 32 |
context = "__START__"
|
| 33 |
else:
|
| 34 |
context = desc_sentences[i-1]
|
| 35 |
-
out = tokenizer(context,
|
| 36 |
max_length = 256,
|
| 37 |
truncation=True,
|
| 38 |
return_attention_mask=True,
|
| 1 |
---
|
| 2 |
language: "en"
|
| 3 |
datasets:
|
| 4 |
+
- spotify-podcast-dataset
|
| 5 |
tags:
|
| 6 |
- bert
|
| 7 |
- classification
|
| 32 |
context = "__START__"
|
| 33 |
else:
|
| 34 |
context = desc_sentences[i-1]
|
| 35 |
+
out = tokenizer(context, s, padding = "max_length",
|
| 36 |
max_length = 256,
|
| 37 |
truncation=True,
|
| 38 |
return_attention_mask=True,
|