|
@@ -0,0 +1,43 @@ |
|
|
|
|
|
import os |
|
|
|
|
|
import unittest |
|
|
|
|
|
|
|
|
|
|
|
from fastNLP.core.preprocess import SeqLabelPreprocess |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestSeqLabelPreprocess(unittest.TestCase): |
|
|
|
|
|
def test_case_1(self): |
|
|
|
|
|
data = [ |
|
|
|
|
|
[['Tom', 'and', 'Jerry', '.'], ['n', '&', 'n', '.']], |
|
|
|
|
|
[['Hello', 'world', '!'], ['a', 'n', '.']], |
|
|
|
|
|
[['Tom', 'and', 'Jerry', '.'], ['n', '&', 'n', '.']], |
|
|
|
|
|
[['Hello', 'world', '!'], ['a', 'n', '.']], |
|
|
|
|
|
[['Tom', 'and', 'Jerry', '.'], ['n', '&', 'n', '.']], |
|
|
|
|
|
[['Hello', 'world', '!'], ['a', 'n', '.']], |
|
|
|
|
|
[['Tom', 'and', 'Jerry', '.'], ['n', '&', 'n', '.']], |
|
|
|
|
|
[['Hello', 'world', '!'], ['a', 'n', '.']], |
|
|
|
|
|
[['Tom', 'and', 'Jerry', '.'], ['n', '&', 'n', '.']], |
|
|
|
|
|
[['Hello', 'world', '!'], ['a', 'n', '.']], |
|
|
|
|
|
] |
|
|
|
|
|
|
|
|
|
|
|
if os.path.exists("./save"): |
|
|
|
|
|
for root, dirs, files in os.walk("./save", topdown=False): |
|
|
|
|
|
for name in files: |
|
|
|
|
|
os.remove(os.path.join(root, name)) |
|
|
|
|
|
for name in dirs: |
|
|
|
|
|
os.rmdir(os.path.join(root, name)) |
|
|
|
|
|
result = SeqLabelPreprocess().run(train_dev_data=data, train_dev_split=0.4, |
|
|
|
|
|
pickle_path="./save") |
|
|
|
|
|
result = SeqLabelPreprocess().run(train_dev_data=data, train_dev_split=0.4, |
|
|
|
|
|
pickle_path="./save") |
|
|
|
|
|
if os.path.exists("./save"): |
|
|
|
|
|
for root, dirs, files in os.walk("./save", topdown=False): |
|
|
|
|
|
for name in files: |
|
|
|
|
|
os.remove(os.path.join(root, name)) |
|
|
|
|
|
for name in dirs: |
|
|
|
|
|
os.rmdir(os.path.join(root, name)) |
|
|
|
|
|
result = SeqLabelPreprocess().run(test_data=data, train_dev_data=data, |
|
|
|
|
|
pickle_path="./save", train_dev_split=0.4, |
|
|
|
|
|
cross_val=True) |
|
|
|
|
|
result = SeqLabelPreprocess().run(test_data=data, train_dev_data=data, |
|
|
|
|
|
pickle_path="./save", train_dev_split=0.4, |
|
|
|
|
|
cross_val=True) |