@@ -299,8 +299,8 @@ class CWS(API): | |||||
te_dataset = reader.load(filepath) | te_dataset = reader.load(filepath) | ||||
pp(te_dataset) | pp(te_dataset) | ||||
from fastNLP.core.tester import Tester | |||||
from fastNLP.core.metrics import BMESF1PreRecMetric | |||||
from ..core.tester import Tester | |||||
from ..core.metrics import BMESF1PreRecMetric | |||||
tester = Tester(data=te_dataset, model=cws_model, metrics=BMESF1PreRecMetric(target='target'), batch_size=64, | tester = Tester(data=te_dataset, model=cws_model, metrics=BMESF1PreRecMetric(target='target'), batch_size=64, | ||||
verbose=0) | verbose=0) | ||||
@@ -232,7 +232,7 @@ class SeqLenProcessor(Processor): | |||||
return dataset | return dataset | ||||
from fastNLP.core.utils import _build_args | |||||
from ..core.utils import _build_args | |||||
class ModelProcessor(Processor): | class ModelProcessor(Processor): | ||||
@@ -1,6 +1,6 @@ | |||||
from functools import wraps | from functools import wraps | ||||
from collections import Counter | from collections import Counter | ||||
from fastNLP.core.dataset import DataSet | |||||
from .dataset import DataSet | |||||
def _check_build_vocab(func): | def _check_build_vocab(func): | ||||
"""A decorator to make sure the indexing is built before used. | """A decorator to make sure the indexing is built before used. | ||||