Sample code for tf.data.generator and fitDataset() of tensorflow.js
This is for someone who has large training dataset and that dataset doesn't fit entirely in memory.
Needs 3steps to see how it works
$ tsc
$ node dist/create-datafiles.js
$ node dist/index.js
- Build this typescript source
- create test dataset files
- build model and learn with
tf.data.generator
andfitDataSet()
Regression model to predict 1 value given 4 arguments value.
This value is caluculated by simple function, named hiddenCalc()