fix bugs for numerical and categorical feature filtering in sklearn template

GoogleCloudPlatform / cloudml-samples

Cloud ML Engine repo. Please visit the new Vertex AI samples repo at https://github.com/GoogleCloudPlatform/vertex-ai-samples

https://cloud.google.com/ai-platform/docs/

Apache License 2.0

1.52k stars 857 forks source link

fix bugs for numerical and categorical feature filtering in sklearn template #417

Closed luotigerlsx closed 5 years ago

luotigerlsx commented 5 years ago

This PR is to fix the bugs in sklearn template. The previous version will raise exception for the case the there is no certain type of features, e.g. numerical or categorical.

This change is

luotigerlsx commented 5 years ago

Hi @gogasca and @cfezequiel , I found a bug in the sklearn template and this PR will fix it. Please help to take a look.

Thanks,

Shixin