Spaces:
Runtime error
Runtime error
Joshua Lansford
commited on
Commit
·
ee0d60a
1
Parent(s):
a97af56
Testing example programs.
Browse files- example_run.py +19 -0
- example_train.py +15 -0
example_run.py
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import pandas as pd
|
| 2 |
+
import transmorgrify
|
| 3 |
+
|
| 4 |
+
#load inference data
|
| 5 |
+
inference_data = pd.read_csv( "phonetics_out_gpu_4000.csv" )
|
| 6 |
+
|
| 7 |
+
#Load the model
|
| 8 |
+
my_model = transmorgrify.Transmorgrifier()
|
| 9 |
+
my_model.load( "my_model.tm" )
|
| 10 |
+
|
| 11 |
+
#do the inference
|
| 12 |
+
#model returns a generator so wrap it with a list
|
| 13 |
+
for result in my_model.execute( inference_data["in_data"] ):
|
| 14 |
+
print( result )
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
# results = list( my_model.execute( inference_data["in_data"] ) )
|
| 18 |
+
|
| 19 |
+
# print( results )
|
example_train.py
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import pandas as pd
|
| 2 |
+
import transmorgrify
|
| 3 |
+
|
| 4 |
+
#load training data
|
| 5 |
+
train_data = pd.read_csv( "phonetics_out_gpu_4000.csv" )[0:100]
|
| 6 |
+
|
| 7 |
+
#do the training
|
| 8 |
+
my_model = transmorgrify.Transmorgrifier()
|
| 9 |
+
my_model.train(
|
| 10 |
+
from_sentances=train_data["in_data"],
|
| 11 |
+
to_sentances=train_data["out_data"],
|
| 12 |
+
iterations=100, )
|
| 13 |
+
|
| 14 |
+
#save the results
|
| 15 |
+
my_model.save( "my_model.tm" )
|