diff --git a/cml/ports/ml_adapter.py b/cml/ports/ml_adapter.py index aa56b8b96c629e73f26f0a59ce7d52affc13e5ac..fcd521238151dee0df0563918ccec8fee7f45198 100644 --- a/cml/ports/ml_adapter.py +++ b/cml/ports/ml_adapter.py @@ -66,6 +66,7 @@ class ConstructionClusteringMLModel(MachineLearningModel): def __init__(self, model): self.__model = model self._cluster = 2 + self.abbreviation = self.__model.__class__.__name__[0:3] def get_labels(self): return self.__model.__getattribute__( @@ -100,6 +101,7 @@ class ReconstructionConceptualMLModel(MachineLearningModel): def __init__(self, model): self.__model = model self.accuracy = None + self.subject = model.__class__.__name__ def train(self, data, *args, **kwargs): # TODO (dmt): Improve signature of this function! @@ -117,6 +119,7 @@ class ReconstructionProceduralMLModel(MachineLearningModel): self.__model = model self.mean_error = None self.max_error = None + self.subject = model.__class__.__name__ def train(self, data, *args, **kwargs): # TODO (dmt): Provide a better way dealing with