googleinterns / amaranth

Apache License 2.0
2 stars 0 forks source link

Ftr/text vectorization ext #35

Closed tommylau-exe closed 4 years ago

tommylau-exe commented 4 years ago

Fixes #36

Perform text vectorization processes externally from Tensorflow.

tommylau-exe commented 4 years ago

Prematurely updated model, undone with latest force-push

tommylau-exe commented 4 years ago

Updated model so that Tokenization file (tokenizer.json) is accurate.

Metrics

Confusion Matrix on held-out set

Predicted Low-Calorie Predicted Average-Calorie Predicted High-Calorie
Actual Low-Calorie 11096 1601 648
Actual Average-Calorie 1673 12788 2264
Actual High-Calorie 782 2298 21196
tommylau-exe commented 4 years ago

Updated model due to further tokenization changes.

Metrics

Confusion Matrix on held-out set

Predicted Low-Calorie Predicted Average-Calorie Predicted High-Calorie
Actual Low-Calorie 11185 1590 609
Actual Average-Calorie 1612 13053 2166
Actual High-Calorie 657 2471 21003