movie_script persona maker

gorkemgoknar · Dec 18, 2020 · f21a449 · f21a449
1 parent 4609ddd
commit f21a449
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/data/make_persona_file.py b/data/make_persona_file.py
@@ -399,6 +399,19 @@ def get_persona_from_file(file):
 
 #personachat["2001"] = list( persona.values() ) 
 
+persona =[]
+for name,val in personachat.items():
+  for p in val:
+    persona.append(p)
+
+
+
+personachat = {}
+train_split_index = round(len(persona)*0.8)
+personachat["train"] = persona[0:train_split_index]
+personachat["val"] = persona[train_split_index:]
+
+
 
 
 ##SAVE TO JSON