for i in range(numberOfEpoch):
if 6 <= i <= 10:
rbm.setMomentum(0.8)
if 11 <= i:
rbm.setMomentum(0.9)
dataLoader.StartNewEpoch()
if i >=40:
learnOneEpoch(verbose=Verbose, T=5)
elif i >=20:
learnOneEpoch(verbose=Verbose, T=3)
else:
learnOneEpoch(verbose=Verbose, T=1)
with open("Outs/"+sys.argv[1]+"_validation_RMSE.txt", "a") as rmsesFile:
dataLoader.StartNewValidationSet()
GetVisiableLayer = dataLoader.GiveVisibleLayerForValidation
setSize = dataLoader.validationSetSize
rmsesFile.write("Epoch {0}, RMSE {1}\n".format(i+1, computeRMSE(verbose=Verbose)))
rmsesFile.flush()
with open("Outs/"+sys.argv[1]+"_training_RMSE.txt", "a") as rmsesFile:
dataLoader.StartNewValidationFromTrainingSet()
GetVisiableLayer = dataLoader.GiveVisibleLayerForValidationFromTraining
setSize = dataLoader.validationFromTrainingSetSize
rmsesFile.write("Epoch {0}, RMSE {1}\n".format(i+1, computeRMSE(verbose=Verbose)))
rmsesFile.flush()
rbm.saveRBM(sys.argv[1])
sys.stdout.flush()
with open(sys.argv[1] + "TRAINING_RMSE.txt", "a") as rmsesFile:
dataLoader.StartNewValidationFromTrainingSet()
GetVisiableLayer = dataLoader.GiveVisibleLayerForValidationFromTraining
setSize = dataLoader.validationFromTrainingSetSize
rmsesFile.write("Epoch {0}, RMSE {1}\n".format(
0, computeRMSE(verbose=True)))
rmsesFile.flush()
for i in range(numberOfEpoch):
if i >= 6:
rbm.setMomentum(0.8)
dataLoader.StartNewEpoch()
learnOneEpoch(verbose=True)
with open(sys.argv[1] + "VALIDATION_RMSE.txt", "a") as rmsesFile:
dataLoader.StartNewValidationSet()
GetVisiableLayer = dataLoader.GiveVisibleLayerForValidation
setSize = dataLoader.validationSetSize
rmsesFile.write("Epoch {0}, RMSE {1}\n".format(
i + 1, computeRMSE(verbose=True)))
rmsesFile.flush()
with open(sys.argv[1] + "TRAINING_RMSE.txt", "a") as rmsesFile:
dataLoader.StartNewValidationFromTrainingSet()
GetVisiableLayer = dataLoader.GiveVisibleLayerForValidationFromTraining
setSize = dataLoader.validationFromTrainingSetSize
rmsesFile.write("Epoch {0}, RMSE {1}\n".format(
i + 1, computeRMSE(verbose=True)))
rmsesFile.flush()
sys.stdout.flush()
rbm.saveRBM()
