kotlindl icon indicating copy to clipboard operation
kotlindl copied to clipboard

Multiple outputs

Open jessiejs opened this issue 11 months ago • 3 comments

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.

jessiejs avatar Mar 02 '24 14:03 jessiejs

Hi @jessiejs could you please your share your case with multiple outputs? What kind of models do you want to build?

zaleslaw avatar Mar 05 '24 14:03 zaleslaw

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 avatar Mar 05 '24 22:03 jessiejs

@jessiejs do you mean to build it with the TF module? Probably it's impossible for now

zaleslaw avatar Mar 19 '24 10:03 zaleslaw