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:
def __init__(self):
pass
self.pipeline = None
self.model = None

def predict(self):
pass

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

Loading…
Cancel
Save