The train method is a function that trains a neural network using a set of training data. It takes in an array of DataSet objects, the number of epochs to train for, and the learning rate. The method modifies the weights and biases of the network to minimize the cost function and improve its accuracy on the training data.

net.train(datasets, 100, 10, 0.01);

The predict method is a function that takes in a Tensor object representing the input to the neural network and returns a Promise that resolves to a Tensor object representing the output of the network. This method is used to make predictions on new data after the network has been trained.

const input = new Tensor([1, 2, 3, 4]);
const output = await net.predict(input);

The save method is a function that saves the network to a Uint8Array. This method is used to save the network after it has been trained.

const buffer = net.save();

The saveFile method is a function that takes in a string representing the path to a file and saves the network to that file. This method is used to save the network after it has been trained.

net.saveFile("model.bin");