Weight Optimizer
Labelprop Index / Labelprop / Weight Optimizer
Auto-generated documentation for labelprop.weight_optimizer module.
MLP
Show source in weight_optimizer.py:14
Signature
class MLP(pl.LightningModule):
def __init__(self, hidden_size=16, learning_rate=1e-05): ...
MLP().configure_optimizers
Show source in weight_optimizer.py:38
Signature
def configure_optimizers(self): ...
MLP().forward
Show source in weight_optimizer.py:22
Signature
def forward(self, x): ...
MLP().training_step
Show source in weight_optimizer.py:28
Signature
def training_step(self, batch, batch_idx): ...
WeightsDataset
Show source in weight_optimizer.py:41
Signature
class WeightsDataset(torch.data.Dataset):
def __init__(self, weights, Y_up, Y_down, Y_true): ...
optimize_weights
Show source in weight_optimizer.py:59
Signature
def optimize_weights(weights, Y_up, Y_down, Y_true, ckpt=None, learning_rate=1e-05): ...