def detectors(seed):
if os.environ.get('CIRCLECI', False):
dets = [
AutoEncoder(num_epochs=1, seed=seed),
DAGMM(num_epochs=1, seed=seed),
DAGMM(num_epochs=1,
autoencoder_type=DAGMM.AutoEncoder.LSTM,
seed=seed),
LSTMAD(num_epochs=1, seed=seed),
LSTMED(num_epochs=1, seed=seed),
RecurrentEBM(num_epochs=1, seed=seed)
]
else:
standard_epochs = 40
dets = [
AutoEncoder(num_epochs=standard_epochs, seed=seed),
DAGMM(num_epochs=standard_epochs, seed=seed, lr=1e-4),
DAGMM(num_epochs=standard_epochs,
autoencoder_type=DAGMM.AutoEncoder.LSTM,
seed=seed),
LSTMAD(num_epochs=standard_epochs, seed=seed),
LSTMED(num_epochs=standard_epochs, seed=seed),
RecurrentEBM(num_epochs=standard_epochs, seed=seed)
]
return sorted(dets, key=lambda x: x.framework)
def detectors(seed):
standard_epochs = 40
dets = [AutoEncoder(num_epochs=standard_epochs, seed=seed),
DAGMM(num_epochs=standard_epochs, seed=seed, lr=1e-4),
DAGMM(num_epochs=standard_epochs, autoencoder_type=DAGMM.AutoEncoder.LSTM, seed=seed),
LSTMAD(num_epochs=standard_epochs, seed=seed), LSTMED(num_epochs=standard_epochs, seed=seed),
RecurrentEBM(num_epochs=standard_epochs, seed=seed)]
return sorted(dets, key=lambda x: x.framework)
def different_window_detectors(seed):
standard_epochs = 40
dets = [LSTMAD(num_epochs=standard_epochs)]
for window_size in [13, 25, 50, 100]:
dets.extend([
LSTMED(name='LSTMED Window: ' + str(window_size),
num_epochs=standard_epochs,
seed=seed,
sequence_length=window_size),
AutoEncoder(name='AE Window: ' + str(window_size),
num_epochs=standard_epochs,
seed=seed,
sequence_length=window_size)
])
return dets