QuadrupleDataset

class pinnx.problem.QuadrupleDataset(X_train, y_train, X_test, y_test, approximator=None, loss_fn='MSE', loss_weights=None)

Dataset with each data point as a quadruple.

The couple of the first three elements are the input, and the fourth element is the output. This dataset can be used with the network MIONet for operator learning.

Parameters:

• X_train – A tuple of three NumPy arrays.

• y_train – A NumPy array.

losses(inputs, outputs, targets, **kwargs)

Return a list of losses, i.e., constraints.

test()

Return a test dataset.

train_next_batch(batch_size=None)

Return a training dataset of the size batch_size.