No classes in regression modelsgit diff

Author: antoinecarme

converters/svm_converter.py

@@ -20,6 +20,8 @@ def get_svm_as_dict(self, clf):
         lDict = {}
         lDict1 = clf.__dict__
         L = clf.support_vectors_.shape[0]
+        if(hasattr(clf, "classes")):
+            lDict["classes"] = len(clf.classes_)
         lDict["L"] = L
         lSVs = {}
         P = int(np.log(L) / np.log(10) + 1)
@@ -47,7 +49,6 @@ def convert_classifier(self, clf):
         lDict1 = clf.__dict__
         lDict["metadata"] = self.get_metadata(clf)
         lDict["options"] = self.get_model_options_as_dict(clf)
-        lDict["classes"] = list(clf.classes_)
         lDict["svm_model"] = self.get_svm_as_dict(clf)
         return lDict