Skip to content

Commit

Permalink
feat: add reference to Autoencoder anomaly detector
Browse files Browse the repository at this point in the history
  • Loading branch information
justinuliu authored and hmgomes committed Jul 29, 2024
1 parent e27cbe9 commit d8787e8
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion src/capymoa/anomaly/autoencoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit d8787e8

Please sign in to comment.