1 files changed, 4 insertions, 6 deletions
diff --git a/ml/ADCharts.h b/ml/ADCharts.h
index 0be324f7d..ee09669e2 100644
--- a/ml/ADCharts.h
+++ b/ml/ADCharts.h
@@ -3,20 +3,18 @@
 #ifndef ML_ADCHARTS_H
 #define ML_ADCHARTS_H
 
+#include "Stats.h"
 #include "ml-private.h"
 
 namespace ml {
 
-void updateDimensionsChart(RRDHOST *RH,
-                           collected_number NumTrainedDimensions,
-                           collected_number NumNormalDimensions,
-                           collected_number NumAnomalousDimensions);
+void updateDimensionsChart(RRDHOST *RH, const MachineLearningStats &MLS);
 
 void updateHostAndDetectionRateCharts(RRDHOST *RH, collected_number AnomalyRate);
 
-void updateDetectionChart(RRDHOST *RH);
+void updateResourceUsageCharts(RRDHOST *RH, const struct rusage &PredictionRU, const struct rusage &TrainingRU);
 
-void updateTrainingChart(RRDHOST *RH, struct rusage *TRU);
+void updateTrainingStatisticsChart(RRDHOST *RH, const TrainingStats &TS);
 
 } // namespace ml