Transfer

class mastml.models.Transfer(prefit_path, nn_params=None, val_size=None, test_size=None, freeze_n_layers=None, cover_model=None, savepath='transfer_model')[source]

Bases: object

Methods Summary

fit(X_train, y_train)

predict(X)

Methods Documentation

fit(X_train, y_train)[source]
predict(X)[source]