googleinterns / amaranth

Apache License 2.0
2 stars 0 forks source link

Refactor NLP Pipeline to use Keras TextVectorization #20

Closed tommylau-exe closed 4 years ago

tommylau-exe commented 4 years ago

The current implementation for the model's NLP pipeline is all custom, and cannot be serialized in it's current form. By transitioning to the Keras TextVectorization layer, although experimental, we'll be able to easily serialize the entirety of the model's logic. This allows us to reload the ML model in other applications, such as the upcoming Chrome extension.