diff --git a/cml/ports/ml_adapter.py b/cml/ports/ml_adapter.py index 5edc450edb0b908932b64efd4a75488324de14b6..e291139d99371986af5612fb3b0d945aa86c1707 100644 --- a/cml/ports/ml_adapter.py +++ b/cml/ports/ml_adapter.py @@ -116,6 +116,7 @@ class ReconstructionConceptualMLModel(MachineLearningModel): @log_warning def train(self, data, *args, **kwargs): # TODO (dmt): Improve signature of this function! + # TODO (dmt): Check if fit can handle data. labels = args[0] self.__model = self.__model.fit(data, labels) self.accuracy = self.__model.score(data, labels)