From 4f734943b11b5f8e9568d69ae8ae3454feec2a8b Mon Sep 17 00:00:00 2001 From: Niels Nuyttens Date: Mon, 17 Jun 2024 14:00:05 +0200 Subject: [PATCH] [skip ci] Support running domain classifier multivariate drift detection from CLI and runner --- nannyml/runner.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/nannyml/runner.py b/nannyml/runner.py index dcba9cc8..709d44dd 100644 --- a/nannyml/runner.py +++ b/nannyml/runner.py @@ -19,6 +19,7 @@ from nannyml.distribution.categorical import CategoricalDistributionCalculator from nannyml.distribution.continuous import ContinuousDistributionCalculator from nannyml.drift.multivariate.data_reconstruction import DataReconstructionDriftCalculator +from nannyml.drift.multivariate.domain_classifier import DomainClassifierCalculator from nannyml.drift.univariate import UnivariateDriftCalculator from nannyml.exceptions import InvalidArgumentsException from nannyml.io import FileReader, FilesystemStore, Writer, WriterFactory @@ -74,7 +75,8 @@ class CalculatorFactory: registry: Dict[str, Type] = { 'univariate_drift': UnivariateDriftCalculator, - 'multivariate_drift': DataReconstructionDriftCalculator, + 'reconstruction_error': DataReconstructionDriftCalculator, + 'domain_classifier': DomainClassifierCalculator, 'performance': PerformanceCalculator, 'cbpe': CBPE, 'dle': DLE,