Finally, there are several columns at the end that are irrelevant for our purposes, so we remove them:
design_cols = ['CSTRATM','CPSUM','PATWT','EDWT']
X_train.drop(design_cols, axis=1, inplace=True)
X_test.drop(design_cols, axis=1, inplace=True)
Finally, there are several columns at the end that are irrelevant for our purposes, so we remove them:
design_cols = ['CSTRATM','CPSUM','PATWT','EDWT']
X_train.drop(design_cols, axis=1, inplace=True)
X_test.drop(design_cols, axis=1, inplace=True)