From dff4cdf6a79d5a6426eaae13ca1235daffc3421b Mon Sep 17 00:00:00 2001 From: xuyige Date: Fri, 9 Nov 2018 22:20:12 +0800 Subject: [PATCH] update API --- fastNLP/api/api.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/fastNLP/api/api.py b/fastNLP/api/api.py index 202f782f..b557038b 100644 --- a/fastNLP/api/api.py +++ b/fastNLP/api/api.py @@ -1,11 +1,16 @@ +import _pickle + class API: def __init__(self): - pass + self.pipeline = None + self.model = None def predict(self): pass - def load(self): - pass \ No newline at end of file + def load(self, name): + _dict = _pickle.load(name) + self.pipeline = _dict['pipeline'] + self.model = _dict['model']