Function main

chatbotv3/encoder_decoder_seq2seq.py:349–361

(op)

def main(op):
    np.set_printoptions(threshold='nan')
    lstm = MyLSTM()
    lstm.load_word_vectors()
    lstm.load_one_hot_word_vectors()
    if op == 'train':
        lstm.train()
    elif op == 'predict':
        lstm.predict()
    elif op == 'test':
        lstm.test()
    else:
        print 'Usage:'

if __name__ == '__main__':
    if len(sys.argv) == 2: