allow DI with CNN
This commit is contained in:
parent
71c6ff18c4
commit
2c3a310ce2
@ -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(
|
||||
|
Loading…
Reference in New Issue
Block a user