From d8787e80d4d1dc5527ca5c47c88743751760d6f9 Mon Sep 17 00:00:00 2001 From: Justin Liu Date: Mon, 8 Jul 2024 20:11:28 +1200 Subject: [PATCH] feat: add reference to Autoencoder anomaly detector --- src/capymoa/anomaly/autoencoder.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/capymoa/anomaly/autoencoder.py b/src/capymoa/anomaly/autoencoder.py index d11d6467..3bdd5762 100644 --- a/src/capymoa/anomaly/autoencoder.py +++ b/src/capymoa/anomaly/autoencoder.py @@ -11,7 +11,12 @@ class Autoencoder(AnomalyDetector): """ Autoencoder anomaly detector This is a simple autoencoder anomaly detector that uses a single hidden layer. - The autoencoder is a duplicated version of MOA's Autoencoder class, but written in PyTorch.""" + + Reference: + + Moulton, R. H., Viktor, H. L., Japkowicz, N., & Gama, J. (2019). + Contextual One-Class Classification in Data Streams. arXiv preprint + arXiv:1907.04233.""" def __init__(self, schema=None, hidden_layer=2, learning_rate=0.5, threshold=0.6, random_seed=1): """Construct a Half-Space Trees anomaly detector