Skip to content

Commit df14808

Browse files
committed
Update pmml4s to 1.5.2
1 parent e268cfe commit df14808

11 files changed

+3
-3
lines changed
655 KB
Binary file not shown.
-233 KB
Binary file not shown.
245 KB
Binary file not shown.
-11.1 KB
Binary file not shown.
11.2 KB
Binary file not shown.
-1.92 MB
Binary file not shown.
1.98 MB
Binary file not shown.
288 KB
Binary file not shown.
-290 KB
Binary file not shown.

pypmml_spark/test/test_score_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def test_from_string(self):
4141
out_df = model.transform(df)
4242
out_df.show()
4343
self.assertTrue(len(out_df.schema) == 11)
44-
self.assertTrue(out_df.schema['PredictedValue'])
44+
self.assertTrue(out_df.schema['predicted_class'])
4545

4646
def test_from_file(self):
4747
df = SQLContext(self.sc).read.csv('./resources/data/Iris.csv', header='true', inferSchema='true')
@@ -64,7 +64,7 @@ def test_from_bytes_array(self):
6464
out_df = model.transform(df)
6565
out_df.show()
6666
self.assertTrue(len(out_df.schema) == 6)
67-
self.assertTrue(out_df.schema['PredictedValue'])
67+
self.assertTrue(out_df.schema['predicted_class'])
6868

6969

7070
if __name__ == "__main__":

0 commit comments

Comments
 (0)