noch inator commited on
clarified example
Browse files- ThoughtVectors.py +4 -4
ThoughtVectors.py
CHANGED
|
@@ -708,7 +708,7 @@ class ThoughtVectors:
|
|
| 708 |
return translator
|
| 709 |
|
| 710 |
|
| 711 |
-
# Example usage
|
| 712 |
if __name__ == "__main__":
|
| 713 |
import csv
|
| 714 |
|
|
@@ -720,12 +720,12 @@ if __name__ == "__main__":
|
|
| 720 |
reader = csv.reader(file)
|
| 721 |
for row in reader:
|
| 722 |
if row:
|
| 723 |
-
yield [row[0]]
|
| 724 |
|
| 725 |
|
| 726 |
-
data = data_generator("
|
| 727 |
|
| 728 |
-
#
|
| 729 |
tv = ThoughtVectors()
|
| 730 |
tv.train(
|
| 731 |
group_data=data,
|
|
|
|
| 708 |
return translator
|
| 709 |
|
| 710 |
|
| 711 |
+
# Example usage
|
| 712 |
if __name__ == "__main__":
|
| 713 |
import csv
|
| 714 |
|
|
|
|
| 720 |
reader = csv.reader(file)
|
| 721 |
for row in reader:
|
| 722 |
if row:
|
| 723 |
+
yield [row[0]] # assuming first collumn is sentences
|
| 724 |
|
| 725 |
|
| 726 |
+
data = data_generator("path/to/csv")
|
| 727 |
|
| 728 |
+
# ThoughtVectors training and usage process
|
| 729 |
tv = ThoughtVectors()
|
| 730 |
tv.train(
|
| 731 |
group_data=data,
|