kotlindl
kotlindl copied to clipboard
Multiple outputs
The only helpful dataset for custom data I can find is OnHeapDataset
, but it appears to only take in multiple inputs but one output. Is there any way to make a dataset for a model that takes in an array of floats and returns an array of floats.
Hi @jessiejs could you please your share your case with multiple outputs? What kind of models do you want to build?
I would like to build a text completion model that takes in one hot vectors for tokens, and outputs a one hot vector for the next token.
@jessiejs do you mean to build it with the TF module? Probably it's impossible for now