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" )
|