From 88ef4da496d848f3d150967748de61d9a604fefa Mon Sep 17 00:00:00 2001
From: Mahdi Dibaiee <mdibaiee@aol.com>
Date: Sat, 25 May 2019 12:09:34 +0430
Subject: [PATCH] fix(model): class_weight check using .any

---
 biomes/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/biomes/model.py b/biomes/model.py
index 8ef0ae7..12dd1b3 100644
--- a/biomes/model.py
+++ b/biomes/model.py
@@ -121,7 +121,7 @@ class Model():
         # map_callback = MapHistory()
 
         extra_params = {}
-        if self.class_weight:
+        if self.class_weight.any():
             extra_params['class_weight'] = self.class_weight
 
         out = self.model.fit(