Methods
# inner loadMnist(trainingSize, testSize) → {String}
Loads the MNIST dataset with specified training and test set sizes.
Parameters:
Name | Type | Description |
---|---|---|
trainingSize |
Number
|
The size of training dataset |
testSize |
Number
|
The size of test dataset |
load_mnist_dataset(trainingSize, testSize)
String
# inner loadMnistModel() → {String}
Loads a pre-trained MNIST model from a specified URL.
await tf.loadLayersModel('https://raw.githubusercontent.com/Cheeetah97/mnist_pretrained_model/main/model.json')
String