def input_fn():
doc1 = Document()
doc1.id = 1
doc1.text = 'title: this is mode1 from doc1, body: this is mode2 from doc1'
doc2 = Document()
doc2.id = 2
doc2.text = 'title: this is mode1 from doc2, body: this is mode2 from doc2'
doc3 = Document()
doc3.id = 3
doc3.text = 'title: this is mode1 from doc3, body: this is mode2 from doc3'
return [doc1, doc2, doc3]
def input_fn():
doc1 = Document()
doc1.id = 1
doc1.embedding.CopyFrom(array2pb(np.random.random([7])))
c = doc1.chunks.add()
c.id = 3
c.embedding.CopyFrom(array2pb(np.random.random([5])))
doc2 = Document()
doc2.id = 2
doc2.embedding.CopyFrom(array2pb(np.random.random([3])))
d = doc2.chunks.add()
d.id = 4
d.embedding.CopyFrom(array2pb(np.random.random([9])))
return [doc1, doc2]
def input_fn():
doc1 = Document()
doc1.id = 1
doc1.embedding.CopyFrom(array2pb(e1))
c = doc1.chunks.add()
c.id = 3
c.embedding.CopyFrom(array2pb(e2))
doc2 = Document()
doc2.id = 2
doc2.embedding.CopyFrom(array2pb(e3))
d = doc2.chunks.add()
d.id = 4
d.embedding.CopyFrom(array2pb(e4))
return [doc1, doc2]
