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