training_data = train_data['Close'].values
test_data = test_data['Close'].values