Extract feature columns from train/test dataset
extract_features(data)
data
data with feature columns and others
data containing only feature columns