AlexanderLavelle

I believe this may be caused also by a prediction of a NaN or Inf... I have had some 'letting up' of the issue by using a scaling technique before...

@roseayeon @gbaned I looked at the [tests](https://github.com/tensorflow/tfx/blob/915170c8c3f1762e66117118ee44704755d48e22/tfx/extensions/google_cloud_big_query/example_gen/executor_test.py#L169-L178), and I think I will need to run this one to confirm the following pseudocode: ```python executive_properties['num_shards'] = 5 ... # Check BigQuery...

@gbaned unfortunately I have not had time to run the unit test as posited -- however, this generally just allows the user to reach an argument provided by Apache Beam...

@meethariprasad ``` brute_force = tfrs.layers.factorized_top_k.BruteForce(model._query_model) brute_force.index_from_dataset( movies.batch(128).map(lambda title: (title, model._candidate_model(title))) ) user_batch = [batch for batch in cached_train.take(1)][0] _, titles = brute_force(user_batch, k=3) titles ``` This is a solution to...

@lenassero depending on your desired setup there are a few things you could try by editing the default Docker container you're referring to. If I run `gcloud container images list...

I have been out of the country but am still interested in pursuing this commit.

@lego0901 - I have a potential recommendation for this. What if during the encoding everything was turned into a sequence example (where appropriate) anyway? This should not affect the majority...

@gbaned I have thought about this PR more and it can be improved even beyond here. Part of it is connecting other big query column types upstream, but fundamentally I...

@gbaned no update on my side -- I have used this and confirm it works when intending to use SequenceExamples in TFX downstream components (Transform, Trainer)

@keerthanakadiri Hello! I have brought in the new changes without conflicts. I am currently on PTO but can run tests in a week or early the following week