Closed Motech2 closed 1 year ago
brute_force_layer = tfrs.layers.factorized_top_k.BruteForce( movielens_retrieval_model.query_model )
brute_force_layer.index( candidates_corpus_dataset.batch(100).map( movielens_retrieval_model.candidate_model ), candidates_corpus_dataset )
ValueError: Attempt to convert a value (<MapDataset element_spec=TensorSpec(shape=(None, 32), dtype=tf.float32, name=None)>) with an unsupported type (<class 'tensorflow.python.data.ops.dataset_ops.MapDataset'>) to a Tensor.
brute_force_layer = tfrs.layers.factorized_top_k.BruteForce( movielens_retrieval_model.query_model )
brute_force_layer.index( candidates_corpus_dataset.batch(100).map( movielens_retrieval_model.candidate_model ), candidates_corpus_dataset )
ValueError: Attempt to convert a value (<MapDataset element_spec=TensorSpec(shape=(None, 32), dtype=tf.float32, name=None)>) with an unsupported type (<class 'tensorflow.python.data.ops.dataset_ops.MapDataset'>) to a Tensor.