disable inlcude mask for RF model

HelmholtzAI-Consultants-Munich · Jan 31, 2024 · e37b65c · e37b65c
1 parent f538388
commit e37b65c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/server/dcp_server/models.py b/src/server/dcp_server/models.py
@@ -270,6 +270,8 @@ def __init__(self, model_config, train_config, eval_config, model_name):
             self.classifier = CellClassifierShallowModel(self.model_config,
                                                          self.train_config,
                                                          self.eval_config)
+            # make sure include mask is set to False if we are using the random forest model 
+            self.include_mask = False 
 
     def update_configs(self, train_config, eval_config):
         self.train_config = train_config