allow DI with CNN

This commit is contained in:
robcaulk 2022-12-07 20:30:13 +01:00
parent 71c6ff18c4
commit 2c3a310ce2

View File

@ -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(