Why was TFR-Bert removed?

[Mitchell-gitrepo]

It looks like TFR-Bert based ranking was removed from tensorflow_ranking/extension. Is there a specific reason for this?

[xuanhuiwang]

We have a new version of TFR-BERT based on Keras + Orbit. You can find the new one here: https://github.com/tensorflow/ranking/blob/master/tensorflow_ranking/examples/keras/tfrbert_antique_train.py.

The new version is better and in native keras. Maybe you can try it out.

[vitalyli]

Hi @xuanhuiwang, is there some example how to transition/convert from column based estimator and into Keras? I'm looking to convert tf_ranking_tfrecord.py setup with embedding columns, estimator etc. and into Keras, next step being to use Bert for text instead of my implementation of BPE column based embedding.