def main(epochs):
xTrain, yTrain, xTest, yTest, scaler = createInputs(factors=False)
results, yPred = run(epochs, 2, xTrain, yTrain, xTest, yTest)
results['unscaledMse'] = mean_squared_error(
scaler.inverse_transform(yTest.reshape((-1, 1))),
scaler.inverse_transform(yPred.reshape((-1, 1))))
saveResults("LSTM-ROLLING-FACTORS", epochs, 2, results)
def main(epochs):
xTrain, yTrain, xTest, yTest, scaler = createInputs(factors=False,
type='dense')
results, yPred = run(epochs, 3, xTrain, yTrain, xTest, yTest)
results['unscaledMse'] = mean_squared_error(
scaler.inverse_transform(yTest.reshape((-1, 1))),
scaler.inverse_transform(yPred))
saveResults("DENSE-NOROLLING-NOFACTORS", epochs, 3, results)
def main(epochs):
xTrain, yTrain, xTest, yTest, scaler = createInputs(factors=False,
tripleBarrier=True)
yTrainBinary = to_categorical(yTrain)
yTestBinary = to_categorical(yTest)
results, yPred, model = run(epochs, 1, xTrain, yTrainBinary, xTest,
yTestBinary)
saveResults("LSTM-NOFACTORS-TRIPLEBARRIER", epochs, 2, results)
