tdubon commited on
Commit
a7c207f
1 Parent(s): 9f87a4e

Update tfdecisiontrees_final.py

Browse files
Files changed (1) hide show
  1. tfdecisiontrees_final.py +1 -1
tfdecisiontrees_final.py CHANGED
@@ -88,7 +88,7 @@ print(f"Label classes: {classes}")
88
 
89
  #rename columns containing invalid characters
90
  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"})
91
- 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"})
92
 
93
  #convert from string to integers
94
  # This stage is necessary if your classification label is represented as a
 
88
 
89
  #rename columns containing invalid characters
90
  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"})
91
+ 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"})
92
 
93
  #convert from string to integers
94
  # This stage is necessary if your classification label is represented as a