Skip to content
Snippets Groups Projects
Commit 0fd5c606 authored by dmt's avatar dmt
Browse files

Add TODO for dmt.

parent 77e00eb7
No related branches found
No related tags found
No related merge requests found
...@@ -116,6 +116,7 @@ class ReconstructionConceptualMLModel(MachineLearningModel): ...@@ -116,6 +116,7 @@ class ReconstructionConceptualMLModel(MachineLearningModel):
@log_warning @log_warning
def train(self, data, *args, **kwargs): def train(self, data, *args, **kwargs):
# TODO (dmt): Improve signature of this function! # TODO (dmt): Improve signature of this function!
# TODO (dmt): Check if fit can handle data.
labels = args[0] labels = args[0]
self.__model = self.__model.fit(data, labels) self.__model = self.__model.fit(data, labels)
self.accuracy = self.__model.score(data, labels) self.accuracy = self.__model.score(data, labels)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment