fix(model): class_weight check using .any

This commit is contained in:
Mahdi Dibaiee 2019-05-25 12:09:34 +04:30
parent 8ffeff179f
commit 88ef4da496

View File

@ -121,7 +121,7 @@ class Model():
# map_callback = MapHistory() # map_callback = MapHistory()
extra_params = {} extra_params = {}
if self.class_weight: if self.class_weight.any():
extra_params['class_weight'] = self.class_weight extra_params['class_weight'] = self.class_weight
out = self.model.fit( out = self.model.fit(