fix(model): class_weight check using .any
This commit is contained in:
parent
8ffeff179f
commit
88ef4da496
@ -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(
|
||||||
|
Loading…
Reference in New Issue
Block a user