SentenceTransmorgrifier / example_train.py
Joshua Lansford
Fixing spelling of sentance to sentence in files.
fbbf27f
raw
history blame contribute delete
360 Bytes
import pandas as pd
import transmorgrify
#load training data
train_data = pd.read_csv( "phonetics_out_gpu_4000.csv" )[0:100]
#do the training
my_model = transmorgrify.Transmorgrifier()
my_model.train(
from_sentences=train_data["in_data"],
to_sentences=train_data["out_data"],
iterations=100, )
#save the results
my_model.save( "my_model.tm" )