allow DI with CNN
This commit is contained in:
		| @@ -22,9 +22,9 @@ class BaseTensorFlowModel(IFreqaiModel): | ||||
|     def __init__(self, **kwargs): | ||||
|         super().__init__(config=kwargs['config']) | ||||
|         self.keras = True | ||||
|         if self.ft_params.get("DI_threshold", 0): | ||||
|             self.ft_params["DI_threshold"] = 0 | ||||
|             logger.warning("DI threshold is not configured for Keras models yet. Deactivating.") | ||||
|         # if self.ft_params.get("DI_threshold", 0): | ||||
|         #     self.ft_params["DI_threshold"] = 0 | ||||
|         #     logger.warning("DI threshold is not configured for Keras models yet. Deactivating.") | ||||
|         self.dd.model_type = 'keras' | ||||
|  | ||||
|     def train( | ||||
|   | ||||
		Reference in New Issue
	
	Block a user