From 8dbf0e8699d0426711e2865ab6612bf29299a114 Mon Sep 17 00:00:00 2001 From: s434732 Date: Sat, 15 May 2021 17:09:20 +0200 Subject: [PATCH] Scared --- mongoObserver.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mongoObserver.py b/mongoObserver.py index a6a8ecd..724e691 100644 --- a/mongoObserver.py +++ b/mongoObserver.py @@ -28,7 +28,7 @@ class LogisticRegressionModel(nn.Module): @ex.capture -def readAndtrain(epchos, batch_size, _run): +def readAndtrain(epochs, batch_size, _run): train = pd.read_csv("train.csv") test = pd.read_csv("test.csv") @@ -45,7 +45,7 @@ def readAndtrain(epchos, batch_size, _run): input_dim = 11 output_dim = 1 - _run.info("Batch: " + str(batch_size) + " epoch: " + epchos) + _run.info("Batch: " + str(batch_size) + " epoch: " + epochs) model = LogisticRegressionModel(input_dim, output_dim) model.load_state_dict(torch.load('DEATH_EVENT.pth')) @@ -53,7 +53,7 @@ def readAndtrain(epchos, batch_size, _run): criterion = torch.nn.BCELoss(reduction='mean') optimizer = torch.optim.SGD(model.parameters(), lr=learning_rate) - for epoch in range(epchos): + for epoch in range(epochs): # print ("Epoch #",epoch) model.train() optimizer.zero_grad()