Spaces:
Sleeping
Sleeping
smhavens
commited on
Commit
·
940deb0
1
Parent(s):
fe5d9d2
test again
Browse files
app.py
CHANGED
@@ -63,11 +63,11 @@ def training():
|
|
63 |
# print(example)
|
64 |
train_examples.append(InputExample(texts=[example['text'], example['label']]))
|
65 |
|
66 |
-
|
67 |
|
68 |
|
69 |
|
70 |
-
embeddings = finetune(
|
71 |
|
72 |
return (dataset['train'].num_rows, type(dataset['train'][0]), type(dataset['train'][0]['text']), dataset['train'][0], embeddings)
|
73 |
|
|
|
63 |
# print(example)
|
64 |
train_examples.append(InputExample(texts=[example['text'], example['label']]))
|
65 |
|
66 |
+
train_dataloader = DataLoader(train_examples, shuffle=True, batch_size=16)
|
67 |
|
68 |
|
69 |
|
70 |
+
embeddings = finetune(train_dataloader)
|
71 |
|
72 |
return (dataset['train'].num_rows, type(dataset['train'][0]), type(dataset['train'][0]['text']), dataset['train'][0], embeddings)
|
73 |
|