Browse Source

update API

tags/v0.2.0
xuyige 6 years ago
parent
commit
dff4cdf6a7
1 changed files with 8 additions and 3 deletions
  1. +8
    -3
      fastNLP/api/api.py

+ 8
- 3
fastNLP/api/api.py View File

@@ -1,11 +1,16 @@


import _pickle



class API: class API:
def __init__(self): def __init__(self):
pass
self.pipeline = None
self.model = None


def predict(self): def predict(self):
pass pass


def load(self):
pass
def load(self, name):
_dict = _pickle.load(name)
self.pipeline = _dict['pipeline']
self.model = _dict['model']

Loading…
Cancel
Save