renamed to z_zsource

zinggAI · Apr 26, 2023 · 3bbceb5 · 3bbceb5
1 parent 19d02b4
commit 3bbceb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/phases/exportModel.py b/python/phases/exportModel.py
@@ -32,7 +32,7 @@ def main():
 
 def export_data(labelledData, location):
 
-    baseCols = ['z_cluster', 'z_zid', 'z_prediction', 'z_score', 'z_source', 'z_isMatch']
+    baseCols = ['z_cluster', 'z_zid', 'z_prediction', 'z_score', 'z_zsource', 'z_isMatch']
     sourceDataColumns =  [c for c in labelledData.columns if c not in  baseCols]
     additionalTrainingColumns = ['z_cluster','z_isMatch']
     trainingSampleColumns = [*additionalTrainingColumns, *sourceDataColumns]