1 files changed, 46 insertions, 0 deletions
diff --git a/ml/Stats.h b/ml/Stats.h
new file mode 100644
index 00000000..b99bc39d
--- /dev/null
+++ b/ml/Stats.h
@@ -0,0 +1,46 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#ifndef ML_STATS_H
+#define ML_STATS_H
+
+#include "ml-private.h"
+
+namespace ml {
+
+struct MachineLearningStats {
+    size_t NumMachineLearningStatusEnabled;
+    size_t NumMachineLearningStatusDisabledUE;
+    size_t NumMachineLearningStatusDisabledSP;
+
+    size_t NumMetricTypeConstant;
+    size_t NumMetricTypeVariable;
+
+    size_t NumTrainingStatusUntrained;
+    size_t NumTrainingStatusPendingWithoutModel;
+    size_t NumTrainingStatusTrained;
+    size_t NumTrainingStatusPendingWithModel;
+
+    size_t NumAnomalousDimensions;
+    size_t NumNormalDimensions;
+};
+
+struct TrainingStats {
+    struct rusage TrainingRU;
+
+    size_t QueueSize;
+    size_t NumPoppedItems;
+
+    usec_t AllottedUT;
+    usec_t ConsumedUT;
+    usec_t RemainingUT;
+
+    size_t TrainingResultOk;
+    size_t TrainingResultInvalidQueryTimeRange;
+    size_t TrainingResultNotEnoughCollectedValues;
+    size_t TrainingResultNullAcquiredDimension;
+    size_t TrainingResultChartUnderReplication;
+};
+
+} // namespace ml
+
+#endif /* ML_STATS_H */