GoogleCloudPlatform / ml-design-patterns

Source code accompanying O'Reilly book: Machine Learning Design Patterns
Apache License 2.0
1.89k stars 532 forks source link

Chapter 6: Transform #14

Open mshearer0 opened 4 years ago

mshearer0 commented 4 years ago

Had to copy data directory with csv files from https://github.com/GoogleCloudPlatform/training-data-analyst/tree/master/quests/serverlessml/data to run taxifare_fc.ipynb.

Model trained on Tensorflow 2.3 is incompatible with serving from GCP AI Platform, Tensorflow 2.2.0. Prediction fails with input segment_ids[0] expected type int32 != int64. Training on TF2.2.0 predicts ok.