keras-io
/

TF_Decision_Trees

Tabular Classification

Model card Files Files and versions Metrics Training metrics Community

tdubon commited on Feb 18, 2022

Commit

a7c207f

•

1 Parent(s): 9f87a4e

Update tfdecisiontrees_final.py

Files changed (1) hide show

tfdecisiontrees_final.py +1 -1

tfdecisiontrees_final.py CHANGED Viewed

@@ -88,7 +88,7 @@ print(f"Label classes: {classes}")
 #rename columns containing invalid characters
 train_data = train_data.rename(columns={"fill_inc_questionnaire_for_veteran's_admin": "fill_inc_veterans_admin", "migration_code-change_in_msa": "migration_code_chx_in_msa", "migration_code-change_in_reg" : "migration_code_chx_in_reg", "migration_code-move_within_reg" : "migration_code_move_within_reg"})
-test_data = train_data.rename(columns={"fill_inc_questionnaire_for_veteran's_admin": "fill_inc_veterans_admin", "migration_code-change_in_msa": "migration_code_chx_in_msa", "migration_code-change_in_reg" : "migration_code_chx_in_reg", "migration_code-move_within_reg" : "migration_code_move_within_reg"})
 #convert from string to integers
 # This stage is necessary if your classification label is represented as a

 #rename columns containing invalid characters
 train_data = train_data.rename(columns={"fill_inc_questionnaire_for_veteran's_admin": "fill_inc_veterans_admin", "migration_code-change_in_msa": "migration_code_chx_in_msa", "migration_code-change_in_reg" : "migration_code_chx_in_reg", "migration_code-move_within_reg" : "migration_code_move_within_reg"})
+test_data = test_data.rename(columns={"fill_inc_questionnaire_for_veteran's_admin": "fill_inc_veterans_admin", "migration_code-change_in_msa": "migration_code_chx_in_msa", "migration_code-change_in_reg" : "migration_code_chx_in_reg", "migration_code-move_within_reg" : "migration_code_move_within_reg"})
 #convert from string to integers
 # This stage is necessary if your classification label is represented as a