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