def post(self):
args = predict_parser.parse_args()
input_question = args.question
input_file = "../Data/prediction_api_input"
file = open(input_file, "w")
file.write(input_question)
file.close()
return s2s.predict_seq2seq(input_file, '../Data/vocab_map',
'../model/seq2seq', 'API')
def main():
print("STARTED \n")
if len(sys.argv) > 1:
mode = sys.argv[1]
else:
raise ValueError("Main mode option not provided")
tf.logging._logger.setLevel(logging.INFO)
if mode.upper() == "TRAINING":
#FOR TRAINING
print("Training Started")
s2s.train_seq2seq('Data/final_question_file', 'Data/final_answer_file', 'Data/vocab_map', 'model/seq2seq')
elif mode.upper() == "INFERENCE":
if len(sys.argv) > 2:
pass
else:
sys.argv.append("COMMAND")
#FOR PREDICTION --- 'file'/'command'
infer_mode = sys.argv[2]
if infer_mode.upper() == "FILE":
print("Entered Inference File Mode")
s2s.predict_seq2seq('Data/prediction_input','Data/vocab_map', 'model/seq2seq', 'FILE')
elif infer_mode.upper() == "COMMAND":
print("Entered Inference Command Mode")
command_line_input = input("Question: ")
ans = s2s.predict_seq2seq('Data/prediction_input','Data/vocab_map', 'model/seq2seq', 'COMMAND', None,
command_line_input)
print("\nQuestion: ", command_line_input)
print("\nAnswer: ", ans.replace('<EOS>',''))
else:
raise ValueError("Correct Inference mode (FILE/COMMAND) was not supplied")
elif mode.upper() == 'TESTING':
s2s.predict_seq2seq('Data/testing_input_file', 'Data/vocab_map', 'model/seq2seq', 'TESTING',
'Data/testing_ref_file')
else:
raise ValueError("Correct Main mode (Training/Inference) was not supplied")
print("\nFINISHED \n")