noch inator commited on
Commit
165397d
·
verified ·
1 Parent(s): ba85643

clarified example

Browse files
Files changed (1) hide show
  1. ThoughtVectors.py +4 -4
ThoughtVectors.py CHANGED
@@ -708,7 +708,7 @@ class ThoughtVectors:
708
  return translator
709
 
710
 
711
- # Example usage with a shorter dataset
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("data.csv")
727
 
728
- # Define the ThoughtVectors training and usage process
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,