1 files changed, 45 insertions, 0 deletions
diff --git a/ml/Config.h b/ml/Config.h
new file mode 100644
index 000000000..f29bae3a6
--- /dev/null
+++ b/ml/Config.h
@@ -0,0 +1,45 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef ML_CONFIG_H
+#define ML_CONFIG_H
+
+#include "ml-private.h"
+
+namespace ml {
+
+class Config {
+public:
+    bool EnableAnomalyDetection;
+
+    unsigned MaxTrainSamples;
+    unsigned MinTrainSamples;
+    unsigned TrainEvery;
+
+    unsigned DiffN;
+    unsigned SmoothN;
+    unsigned LagN;
+
+    unsigned MaxKMeansIters;
+
+    double DimensionAnomalyScoreThreshold;
+    double HostAnomalyRateThreshold;
+
+    double ADMinWindowSize;
+    double ADMaxWindowSize;
+    double ADIdleWindowSize;
+    double ADWindowRateThreshold;
+    double ADDimensionRateThreshold;
+
+    SIMPLE_PATTERN *SP_HostsToSkip;
+    SIMPLE_PATTERN *SP_ChartsToSkip;
+
+    std::string AnomalyDBPath;
+
+    void readMLConfig();
+};
+
+extern Config Cfg;
+
+} // namespace ml
+
+#endif /* ML_CONFIG_H */