kubeflow
/
examples
mirror of https://github.com/kubeflow/examples.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Rename issue_summarization.py to IssueSummarization.py (#68) 

* Rename issue_summarization.py to IssueSummarization.py

* The module name is supposed to be the same as the class name
* Fix the predict method signature

* Fix lint
This commit is contained in:
Ankush Agarwal 2018-04-06 21:40:08 -07:00 committed by k8s-ci-robot
parent a4576e48f1
commit e3b826a5af
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
github_issue_summarization/notebooks/issue_summarization.py → github_issue_summarization/notebooks/IssueSummarization.py
View File

@ -21,5 +21,5 @@ class IssueSummarization(object):
decoder_preprocessor=title_pp,
seq2seq_model=load_model('seq2seq_model_tutorial.h5'))
def predict(self, input_text):
def predict(self, input_text, feature_names): # pylint: disable=unused-argument
return np.asarray([[self.model.generate_issue_title(body[0])[1]] for body in input_text])