DJINN: Deep Jointly-Informed Neural Networks’s documentation!
Building djinn regression models with PyTorch and sklearn.
Example
# Basic usage: fit a regression model & predict something new
from djinn import DJINN_Regressor
model = DJINN_Regressor()
model.fit(X,y)
y_new = model.predict(x_new)
# Add more training data with online learning
model.continue_training(X1,y1)
y_new1 = model.predict(x_new)
For more info, see the paper.