Le Yan

Results 4 comments of Le Yan

In general, this should be doable. You could: 1. Use the model to estimator pipeline as in [example](https://github.com/tensorflow/ranking/blob/master/tensorflow_ranking/examples/keras/keras_m2e_tfrecord.py). 2. Or inherit or make your own model_builder and override the [create_inputs()](https://github.com/tensorflow/ranking/blob/77f280075e220946d0935b44b64770f4dabb37d0/tensorflow_ranking/python/keras/model.py#L192)...

Hi Jack, are you just running the `keras_dnn_tfrecord.py` with default parameters? It appears to me a TensorFlow issue. Let me investigate a bit.

Thank you for sharing your issue. This issue is specific to the predict function. There are two quick workarounds if you stick to your model working with full tensors: 1....

@dazcona Thank you for your issue. Based on my reproduce, the error is from the fact that our pipeline works only on the data saved in the ELWC format (treated...