train.set_window(end="2011-04-30")
test.set_window(start="2011-04-30")
train_elec = train.buildings[1].elec
test_elec = test.buildings[1].elec
top_5_train_elec = train_elec.submeters().select_top_k(k=5)
np.random.seed(42)
params = {}
#classifiers = {'CO':CombinatorialOptimisation(), 'FHMM':FHMM()}
predictions = {}
sample_period = 120
co = CombinatorialOptimisation()
fhmm = FHMM()
## Train models
co.train(top_5_train_elec, sample_period=sample_period)
fhmm.train(top_5_train_elec, sample_period=sample_period)
## Export models
co.export_model(filename='co.h5')
fhmm.export_model(filename='fhmm.h5')
co.import_model(filename='co.h5')
fhmm.import_model(filename='fhmm.h5')
