|
|
@ -18,13 +18,12 @@ from optparse import OptionParser
|
|
|
|
from py_paddle import swig_paddle, DataProviderConverter
|
|
|
|
from py_paddle import swig_paddle, DataProviderConverter
|
|
|
|
from paddle.trainer.PyDataProvider2 import sparse_binary_vector
|
|
|
|
from paddle.trainer.PyDataProvider2 import sparse_binary_vector
|
|
|
|
from paddle.trainer.config_parser import parse_config
|
|
|
|
from paddle.trainer.config_parser import parse_config
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
Usage: run following command to show help message.
|
|
|
|
Usage: run following command to show help message.
|
|
|
|
python api_predict.py -h
|
|
|
|
python api_predict.py -h
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class QuickStartPrediction():
|
|
|
|
class QuickStartPrediction():
|
|
|
|
def __init__(self, train_conf, dict_file, model_dir=None, label_file=None):
|
|
|
|
def __init__(self, train_conf, dict_file, model_dir=None, label_file=None):
|
|
|
|
"""
|
|
|
|
"""
|
|
|
@ -72,9 +71,7 @@ class QuickStartPrediction():
|
|
|
|
transform word into integer index according to the dictionary.
|
|
|
|
transform word into integer index according to the dictionary.
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
words = data.strip().split()
|
|
|
|
words = data.strip().split()
|
|
|
|
word_slot = [
|
|
|
|
word_slot = [self.word_dict[w] for w in words if w in self.word_dict]
|
|
|
|
self.word_dict[w] for w in words if w in self.word_dict
|
|
|
|
|
|
|
|
]
|
|
|
|
|
|
|
|
return word_slot
|
|
|
|
return word_slot
|
|
|
|
|
|
|
|
|
|
|
|
def batch_predict(self, data_batch):
|
|
|
|
def batch_predict(self, data_batch):
|
|
|
@ -84,6 +81,7 @@ class QuickStartPrediction():
|
|
|
|
print("predicting labels is:")
|
|
|
|
print("predicting labels is:")
|
|
|
|
print prob
|
|
|
|
print prob
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def option_parser():
|
|
|
|
def option_parser():
|
|
|
|
usage = "python predict.py -n config -w model_dir -d dictionary -i input_file "
|
|
|
|
usage = "python predict.py -n config -w model_dir -d dictionary -i input_file "
|
|
|
|
parser = OptionParser(usage="usage: %s [options]" % usage)
|
|
|
|
parser = OptionParser(usage="usage: %s [options]" % usage)
|
|
|
@ -144,5 +142,6 @@ def main():
|
|
|
|
print labels
|
|
|
|
print labels
|
|
|
|
predict.batch_predict(batch)
|
|
|
|
predict.batch_predict(batch)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if __name__ == '__main__':
|
|
|
|
main()
|
|
|
|
main()
|
|
|
|