thereadme/world-ecoregion
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix(model): class_weight check using .any

Browse Source
This commit is contained in:
Mahdi Dibaiee 2019-05-25 12:09:34 +04:30
parent 8ffeff179f
commit 88ef4da496
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
biomes/model.py
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(