Custom Models

If the standard integrate.ai models (Feedforward Neural Network or Generalized Linear Model) do not suit your needs, you can create a custom model package.

Use the REST API included in the integrate.ai SDK to test and upload a custom model. After you upload it, it can be used by any other users in your workspace.

integrate.ai supports all custom models under pytorch (for example CNNs, LSTMs, Transformers and GLMs).

Customization is restricted in the following ways:

• Models must use one of the following loss functions;

  • Classification, regression, logistic, normal, Poisson, gamma, and inverse gaussian

• Only a single output train function is supported

• Data augmentation is not supported