Merge pull request #1175 from rachitk/nn

Enable multiclass support for TPOT-NN

tpot/builtins/nn.py

@@ -154,17 +154,13 @@ def fit(self, X, y):
     def validate_inputs(self, X, y):
         # Things we don't want to allow until we've tested them:
         # - Sparse inputs
-        # - Multiclass outputs (e.g., more than 2 classes in `y`)
         # - Non-finite inputs
         # - Complex inputs
 
         X, y = check_X_y(X, y, accept_sparse=False, allow_nd=False)
 
         assert_all_finite(X, y)
 
-        if type_of_target(y) != 'binary':
-            raise ValueError("Non-binary targets not supported")
-
         if np.any(np.iscomplex(X)) or np.any(np.iscomplex(y)):
             raise ValueError("Complex data not supported")
         if np.issubdtype(X.dtype, np.object_) or np.issubdtype(y.dtype, np.object_):