Add score output for OneClassSVM model (#101)

* add 2 outs

* update

Author: sneaxiy

sqlflow_models/one_class_svm.py

@@ -103,4 +103,5 @@ def sqlflow_train_loop(self, dataset):
     def sqlflow_predict_one(self, features):
         features = self.concat_features(features)
         pred = self.svm.predict(features)
-        return [pred]
+        score = self.svm.decision_function(features)
+        return pred, score

tests/test_one_class_svm.py

@@ -53,10 +53,10 @@ def test_main(self):
 
         predict_dataset = self.create_dataset()
         for features in dataset_reader(predict_dataset):
-            pred = svm.sqlflow_predict_one(features)
+            pred = svm.sqlflow_predict_one(features)[0]
             pred = np.array(pred)
-            self.assertEqual(pred.shape, (1, 1))
-            self.assertTrue(pred[0][0] == 1 or pred[0][0] == -1)
+            self.assertEqual(pred.shape, (1,))
+            self.assertTrue(pred[0] == 1 or pred[0] == -1)
 
 
 if __name__ == '__main__':