Skip to content
Permalink
Browse files

updated test case

  • Loading branch information...
Nirmal-Neel committed May 14, 2019
1 parent e51312f commit 7213027eeabb0e6543d19cb414df344dd8d45102
Showing with 2 additions and 3 deletions.
  1. +0 −1 .travis.yml
  2. +2 −2 nyoka/tests/skl_to_pmml_UnitTest.py
@@ -1,7 +1,6 @@
language: python
python:
- "3.6"
- "3.7"
cache: pip
install: "pip install git+https://github.com/nyoka-pmml/nyoka.git"
before_script:
@@ -155,7 +155,7 @@ def test_sklearn_06(self):

pipeline_obj.fit(X,y)
skl_to_pmml(pipeline_obj,features,target,f_name)
pmml_obj = pml.parse(f_name,True)
pmml_obj = pml.parse(f_name)

## 1
reg_tab = pmml_obj.RegressionModel[0].RegressionTable[0]
@@ -186,7 +186,7 @@ def test_sklearn_07(self):

pipeline_obj.fit(irisd[features], irisd[target])
skl_to_pmml(pipeline_obj, features, target, f_name)
pmml_obj = pml.parse(f_name,True)
pmml_obj = pml.parse(f_name)

## 1
segmentation = pmml_obj.MiningModel[0].Segmentation

0 comments on commit 7213027

Please sign in to comment.
You can’t perform that action at this time.