Skip to content
This repository was archived by the owner on Apr 15, 2022. It is now read-only.

Commit 1035683

Browse files
authored
Update zeppelin.py
1 parent e0103de commit 1035683

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

splicemachine/ml/zeppelin.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,9 +79,9 @@ def show_confusion_matrix(sc, sqlContext, TP, TN, FP, FN):
7979
:param FP: False Positives
8080
:param FN: False Negatives
8181
"""
82-
confusion_rdd = sc.parallelize([['Predicted', TP, FN], ['Actual', FP, TN]])
82+
confusion_rdd = sc.parallelize([['True', TP, FN], ['False', FP, TN]])
8383
confusion_matrix = sqlContext.createDataFrame(
84-
confusion_rdd, ['', 'Actual', 'Predicted'])
84+
confusion_rdd, ['', 'False', 'True'])
8585
confusion_matrix.show()
8686

8787

0 commit comments

Comments
 (0)