phenonaut.predict.default_predictors.pytorch_models package

Submodules

phenonaut.predict.default_predictors.pytorch_models.dave module

class phenonaut.predict.default_predictors.pytorch_models.dave.DAVE(batch_size=1024, learning_rate=0.001, epochs=10, num_hidden_layers=1, seed: int | None = None)

Bases: object

fit(X, y)

loss_function(recon_XA, recon_XB, XA, XB, muA, muB, logvarA, logvarB)

predict(X)

phenonaut.predict.default_predictors.pytorch_models.multiregressor_nn module

class phenonaut.predict.default_predictors.pytorch_models.multiregressor_nn.MultiRegressorNN(batch_size=128, learning_rate=0.001, epochs=100, num_hidden_layers=1, hidden_layer_sizes: list[int] | None = None, use_optimizer: str = 'ADAM', seed: int | None = None)

Bases: object

fit(X, y)

predict(X)

Module contents