From d4912c0bdcc4ecba96dfd2b643b5e816d51c6bda Mon Sep 17 00:00:00 2001
From: s434766 <przowc@st.amu.edu.pl>
Date: Sun, 16 May 2021 18:02:36 +0200
Subject: [PATCH] sacred my runs

---
 ium_s434766/my_runs/1/config.json             |   6 ++
 ium_s434766/my_runs/1/cout.txt                |   4 +
 ium_s434766/my_runs/1/metrics.json            |   1 +
 ium_s434766/my_runs/1/run.json                |  67 ++++++++++++++
 ium_s434766/my_runs/2/config.json             |   6 ++
 ium_s434766/my_runs/2/cout.txt                |   5 ++
 ium_s434766/my_runs/2/metrics.json            |   1 +
 ium_s434766/my_runs/2/run.json                |  63 +++++++++++++
 ium_s434766/my_runs/3/config.json             |   6 ++
 ium_s434766/my_runs/3/cout.txt                |   4 +
 ium_s434766/my_runs/3/metrics.json            |   1 +
 ium_s434766/my_runs/3/run.json                |  67 ++++++++++++++
 ium_s434766/my_runs/4/config.json             |   6 ++
 ium_s434766/my_runs/4/cout.txt                |   4 +
 ium_s434766/my_runs/4/metrics.json            |   1 +
 ium_s434766/my_runs/4/run.json                |  65 ++++++++++++++
 ium_s434766/my_runs/4/stroke.pth              | Bin 0 -> 1141 bytes
 ...torch1_37f3ae3f09d3a85faa1cb43617a6d59e.py |  85 ++++++++++++++++++
 ...torch1_e0e75cc8f994d35ec0d404b605721131.py |  85 ++++++++++++++++++
 sacred-pytorch1.py                            |   2 +-
 stroke.pth                                    | Bin 1143 -> 1141 bytes
 21 files changed, 478 insertions(+), 1 deletion(-)
 create mode 100644 ium_s434766/my_runs/1/config.json
 create mode 100644 ium_s434766/my_runs/1/cout.txt
 create mode 100644 ium_s434766/my_runs/1/metrics.json
 create mode 100644 ium_s434766/my_runs/1/run.json
 create mode 100644 ium_s434766/my_runs/2/config.json
 create mode 100644 ium_s434766/my_runs/2/cout.txt
 create mode 100644 ium_s434766/my_runs/2/metrics.json
 create mode 100644 ium_s434766/my_runs/2/run.json
 create mode 100644 ium_s434766/my_runs/3/config.json
 create mode 100644 ium_s434766/my_runs/3/cout.txt
 create mode 100644 ium_s434766/my_runs/3/metrics.json
 create mode 100644 ium_s434766/my_runs/3/run.json
 create mode 100644 ium_s434766/my_runs/4/config.json
 create mode 100644 ium_s434766/my_runs/4/cout.txt
 create mode 100644 ium_s434766/my_runs/4/metrics.json
 create mode 100644 ium_s434766/my_runs/4/run.json
 create mode 100644 ium_s434766/my_runs/4/stroke.pth
 create mode 100644 ium_s434766/my_runs/_sources/sacred-pytorch1_37f3ae3f09d3a85faa1cb43617a6d59e.py
 create mode 100644 ium_s434766/my_runs/_sources/sacred-pytorch1_e0e75cc8f994d35ec0d404b605721131.py

diff --git a/ium_s434766/my_runs/1/config.json b/ium_s434766/my_runs/1/config.json
new file mode 100644
index 0000000..9134471
--- /dev/null
+++ b/ium_s434766/my_runs/1/config.json
@@ -0,0 +1,6 @@
+{
+  "batch_size": 16,
+  "learning_rate": 0.001,
+  "num_epochs": 15,
+  "seed": 397973104
+}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/1/cout.txt b/ium_s434766/my_runs/1/cout.txt
new file mode 100644
index 0000000..0a803bb
--- /dev/null
+++ b/ium_s434766/my_runs/1/cout.txt
@@ -0,0 +1,4 @@
+INFO - ium_s434766 - Running command 'my_main'
+INFO - ium_s434766 - Started run with ID "1"
+INFO - train - Batch size = 16 Epochs = 15
+INFO - train - Last loss = 0.32308459281921387
diff --git a/ium_s434766/my_runs/1/metrics.json b/ium_s434766/my_runs/1/metrics.json
new file mode 100644
index 0000000..9e26dfe
--- /dev/null
+++ b/ium_s434766/my_runs/1/metrics.json
@@ -0,0 +1 @@
+{}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/1/run.json b/ium_s434766/my_runs/1/run.json
new file mode 100644
index 0000000..39c8c11
--- /dev/null
+++ b/ium_s434766/my_runs/1/run.json
@@ -0,0 +1,67 @@
+{
+  "artifacts": [],
+  "command": "my_main",
+  "experiment": {
+    "base_dir": "/home/przemek/ium_434766",
+    "dependencies": [
+      "numpy==1.20.1",
+      "pandas==1.2.4",
+      "sacred==0.8.2",
+      "scikit-learn==0.24.1",
+      "torch==1.8.1"
+    ],
+    "mainfile": "sacred-pytorch1.py",
+    "name": "ium_s434766",
+    "repositories": [],
+    "sources": [
+      [
+        "sacred-pytorch1.py",
+        "_sources/sacred-pytorch1_37f3ae3f09d3a85faa1cb43617a6d59e.py"
+      ]
+    ]
+  },
+  "heartbeat": "2021-05-16T16:01:49.590955",
+  "host": {
+    "ENV": {},
+    "cpu": "Intel(R) Core(TM) i5-4690 CPU @ 3.50GHz",
+    "hostname": "OwczarPC",
+    "os": [
+      "Linux",
+      "Linux-5.4.72-microsoft-standard-WSL2-x86_64-with-glibc2.29"
+    ],
+    "python_version": "3.8.5"
+  },
+  "meta": {
+    "command": "my_main",
+    "options": {
+      "--beat-interval": null,
+      "--capture": null,
+      "--comment": null,
+      "--debug": false,
+      "--enforce_clean": false,
+      "--file_storage": null,
+      "--force": false,
+      "--help": false,
+      "--loglevel": null,
+      "--mongo_db": null,
+      "--name": null,
+      "--pdb": false,
+      "--print-config": false,
+      "--priority": null,
+      "--queue": false,
+      "--s3": null,
+      "--sql": null,
+      "--tiny_db": null,
+      "--unobserved": false,
+      "COMMAND": null,
+      "UPDATE": [],
+      "help": false,
+      "with": false
+    }
+  },
+  "resources": [],
+  "result": null,
+  "start_time": "2021-05-16T16:01:49.507065",
+  "status": "COMPLETED",
+  "stop_time": "2021-05-16T16:01:49.589890"
+}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/2/config.json b/ium_s434766/my_runs/2/config.json
new file mode 100644
index 0000000..de842b5
--- /dev/null
+++ b/ium_s434766/my_runs/2/config.json
@@ -0,0 +1,6 @@
+{
+  "batch_size": 16,
+  "learning_rate": 0.001,
+  "num_epochs": 15,
+  "seed": 9915532
+}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/2/cout.txt b/ium_s434766/my_runs/2/cout.txt
new file mode 100644
index 0000000..89c3439
--- /dev/null
+++ b/ium_s434766/my_runs/2/cout.txt
@@ -0,0 +1,5 @@
+INFO - ium_s434766 - Running command 'my_main'
+INFO - ium_s434766 - Started run with ID "2"
+INFO - train - Batch size = 16 Epochs = 15
+INFO - train - Last loss = 0.29651644825935364
+INFO - ium_s434766 - Completed after 0:00:00
diff --git a/ium_s434766/my_runs/2/metrics.json b/ium_s434766/my_runs/2/metrics.json
new file mode 100644
index 0000000..9e26dfe
--- /dev/null
+++ b/ium_s434766/my_runs/2/metrics.json
@@ -0,0 +1 @@
+{}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/2/run.json b/ium_s434766/my_runs/2/run.json
new file mode 100644
index 0000000..d05976f
--- /dev/null
+++ b/ium_s434766/my_runs/2/run.json
@@ -0,0 +1,63 @@
+{
+  "artifacts": [],
+  "command": "my_main",
+  "experiment": {
+    "base_dir": "/home/przemek/ium_434766",
+    "dependencies": [
+      "numpy==1.20.1",
+      "pandas==1.2.4",
+      "sacred==0.8.2",
+      "scikit-learn==0.24.1",
+      "torch==1.8.1"
+    ],
+    "mainfile": "sacred-pytorch1.py",
+    "name": "ium_s434766",
+    "repositories": [],
+    "sources": [
+      [
+        "sacred-pytorch1.py",
+        "_sources/sacred-pytorch1_37f3ae3f09d3a85faa1cb43617a6d59e.py"
+      ]
+    ]
+  },
+  "heartbeat": "2021-05-16T16:01:49.672141",
+  "host": {
+    "ENV": {},
+    "cpu": "Intel(R) Core(TM) i5-4690 CPU @ 3.50GHz",
+    "hostname": "OwczarPC",
+    "os": [
+      "Linux",
+      "Linux-5.4.72-microsoft-standard-WSL2-x86_64-with-glibc2.29"
+    ],
+    "python_version": "3.8.5"
+  },
+  "meta": {
+    "command": "my_main",
+    "options": {
+      "--beat-interval": null,
+      "--capture": null,
+      "--comment": null,
+      "--debug": false,
+      "--enforce_clean": false,
+      "--file_storage": null,
+      "--force": false,
+      "--help": false,
+      "--loglevel": null,
+      "--mongo_db": null,
+      "--name": null,
+      "--pdb": false,
+      "--print-config": false,
+      "--priority": null,
+      "--queue": false,
+      "--s3": null,
+      "--sql": null,
+      "--tiny_db": null,
+      "--unobserved": false
+    }
+  },
+  "resources": [],
+  "result": null,
+  "start_time": "2021-05-16T16:01:49.643954",
+  "status": "COMPLETED",
+  "stop_time": "2021-05-16T16:01:49.670359"
+}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/3/config.json b/ium_s434766/my_runs/3/config.json
new file mode 100644
index 0000000..de41c69
--- /dev/null
+++ b/ium_s434766/my_runs/3/config.json
@@ -0,0 +1,6 @@
+{
+  "batch_size": 16,
+  "learning_rate": 0.001,
+  "num_epochs": 15,
+  "seed": 400864859
+}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/3/cout.txt b/ium_s434766/my_runs/3/cout.txt
new file mode 100644
index 0000000..307075e
--- /dev/null
+++ b/ium_s434766/my_runs/3/cout.txt
@@ -0,0 +1,4 @@
+INFO - ium_s434766 - Running command 'my_main'
+INFO - ium_s434766 - Started run with ID "3"
+INFO - train - Batch size = 16 Epochs = 15
+INFO - train - Last loss = 0.8644022345542908
diff --git a/ium_s434766/my_runs/3/metrics.json b/ium_s434766/my_runs/3/metrics.json
new file mode 100644
index 0000000..9e26dfe
--- /dev/null
+++ b/ium_s434766/my_runs/3/metrics.json
@@ -0,0 +1 @@
+{}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/3/run.json b/ium_s434766/my_runs/3/run.json
new file mode 100644
index 0000000..7b3a97b
--- /dev/null
+++ b/ium_s434766/my_runs/3/run.json
@@ -0,0 +1,67 @@
+{
+  "artifacts": [],
+  "command": "my_main",
+  "experiment": {
+    "base_dir": "/home/przemek/ium_434766",
+    "dependencies": [
+      "numpy==1.20.1",
+      "pandas==1.2.4",
+      "sacred==0.8.2",
+      "scikit-learn==0.24.1",
+      "torch==1.8.1"
+    ],
+    "mainfile": "sacred-pytorch1.py",
+    "name": "ium_s434766",
+    "repositories": [],
+    "sources": [
+      [
+        "sacred-pytorch1.py",
+        "_sources/sacred-pytorch1_e0e75cc8f994d35ec0d404b605721131.py"
+      ]
+    ]
+  },
+  "heartbeat": "2021-05-16T16:02:13.422226",
+  "host": {
+    "ENV": {},
+    "cpu": "Intel(R) Core(TM) i5-4690 CPU @ 3.50GHz",
+    "hostname": "OwczarPC",
+    "os": [
+      "Linux",
+      "Linux-5.4.72-microsoft-standard-WSL2-x86_64-with-glibc2.29"
+    ],
+    "python_version": "3.8.5"
+  },
+  "meta": {
+    "command": "my_main",
+    "options": {
+      "--beat-interval": null,
+      "--capture": null,
+      "--comment": null,
+      "--debug": false,
+      "--enforce_clean": false,
+      "--file_storage": null,
+      "--force": false,
+      "--help": false,
+      "--loglevel": null,
+      "--mongo_db": null,
+      "--name": null,
+      "--pdb": false,
+      "--print-config": false,
+      "--priority": null,
+      "--queue": false,
+      "--s3": null,
+      "--sql": null,
+      "--tiny_db": null,
+      "--unobserved": false,
+      "COMMAND": null,
+      "UPDATE": [],
+      "help": false,
+      "with": false
+    }
+  },
+  "resources": [],
+  "result": null,
+  "start_time": "2021-05-16T16:02:13.390881",
+  "status": "COMPLETED",
+  "stop_time": "2021-05-16T16:02:13.420375"
+}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/4/config.json b/ium_s434766/my_runs/4/config.json
new file mode 100644
index 0000000..ca56d5c
--- /dev/null
+++ b/ium_s434766/my_runs/4/config.json
@@ -0,0 +1,6 @@
+{
+  "batch_size": 16,
+  "learning_rate": 0.001,
+  "num_epochs": 15,
+  "seed": 562374018
+}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/4/cout.txt b/ium_s434766/my_runs/4/cout.txt
new file mode 100644
index 0000000..652bede
--- /dev/null
+++ b/ium_s434766/my_runs/4/cout.txt
@@ -0,0 +1,4 @@
+INFO - ium_s434766 - Running command 'my_main'
+INFO - ium_s434766 - Started run with ID "4"
+INFO - train - Batch size = 16 Epochs = 15
+INFO - train - Last loss = 0.2932703197002411
diff --git a/ium_s434766/my_runs/4/metrics.json b/ium_s434766/my_runs/4/metrics.json
new file mode 100644
index 0000000..9e26dfe
--- /dev/null
+++ b/ium_s434766/my_runs/4/metrics.json
@@ -0,0 +1 @@
+{}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/4/run.json b/ium_s434766/my_runs/4/run.json
new file mode 100644
index 0000000..1db9390
--- /dev/null
+++ b/ium_s434766/my_runs/4/run.json
@@ -0,0 +1,65 @@
+{
+  "artifacts": [
+    "stroke.pth"
+  ],
+  "command": "my_main",
+  "experiment": {
+    "base_dir": "/home/przemek/ium_434766",
+    "dependencies": [
+      "numpy==1.20.1",
+      "pandas==1.2.4",
+      "sacred==0.8.2",
+      "scikit-learn==0.24.1",
+      "torch==1.8.1"
+    ],
+    "mainfile": "sacred-pytorch1.py",
+    "name": "ium_s434766",
+    "repositories": [],
+    "sources": [
+      [
+        "sacred-pytorch1.py",
+        "_sources/sacred-pytorch1_e0e75cc8f994d35ec0d404b605721131.py"
+      ]
+    ]
+  },
+  "heartbeat": "2021-05-16T16:02:13.502327",
+  "host": {
+    "ENV": {},
+    "cpu": "Intel(R) Core(TM) i5-4690 CPU @ 3.50GHz",
+    "hostname": "OwczarPC",
+    "os": [
+      "Linux",
+      "Linux-5.4.72-microsoft-standard-WSL2-x86_64-with-glibc2.29"
+    ],
+    "python_version": "3.8.5"
+  },
+  "meta": {
+    "command": "my_main",
+    "options": {
+      "--beat-interval": null,
+      "--capture": null,
+      "--comment": null,
+      "--debug": false,
+      "--enforce_clean": false,
+      "--file_storage": null,
+      "--force": false,
+      "--help": false,
+      "--loglevel": null,
+      "--mongo_db": null,
+      "--name": null,
+      "--pdb": false,
+      "--print-config": false,
+      "--priority": null,
+      "--queue": false,
+      "--s3": null,
+      "--sql": null,
+      "--tiny_db": null,
+      "--unobserved": false
+    }
+  },
+  "resources": [],
+  "result": null,
+  "start_time": "2021-05-16T16:02:13.476148",
+  "status": "COMPLETED",
+  "stop_time": "2021-05-16T16:02:13.501265"
+}
\ No newline at end of file
diff --git a/ium_s434766/my_runs/4/stroke.pth b/ium_s434766/my_runs/4/stroke.pth
new file mode 100644
index 0000000000000000000000000000000000000000..a6c4e3b885f2746a672f48835f57be55ad6d2d7c
GIT binary patch
literal 1141
zcma)5J#5oJ6h0@ZlRC6X^8@9lMM6SN)sl*|P0}^0BB2{<f{?5(mSdk{gB$nRm#`42
z3@u`2M~I1$8Kqqrn7T8SnU$G4#{}Cbh$lUr?%jLe``*vVIGI8ylR<w;9u<)0>ZUa$
zTZTrp>hW{C-C9J45jjcdx?|f!r<T+AqyyI=E;05kol1b#J0Mh)1qN+feWJP57sTqB
z6cRdh7;;q|P|Nlt)g|45Wg99bea~^#VNHT$rJ|%+pvOG59)Z*(L~7fPMxRWK63i)_
zczt_kcdyx~fv6<eRX-j;8jBP%k1@gm7SCWFi&!`V>5a)bWhq?iz=D$fL#%6Q9xR$H
ztC=zb)LOIAsMmKOXQr`0A@8G|!V(rvp)f%zQYdwxJQ%@sWuBp_M}%s;S#ZP8v|Le`
z6~WCB-15sA64zsmg4<Z|6mgm#?u=l?^o@SeR`1BM3|Jk(UEezJx>XscGY9Ke-=7Hx
zag}FIu8fM2$!*nxQrfLtY<!1n53?5>`Q^#|pKK8j*MA}J#gK#6w_Ce$<j9)Lyp--e
zI{$i=yZG|y{rSe_xADIEVf@2*HRhaFUcYTFGd^5rbIPHODW{;l?bh1=ve}k+rptG|
zrwNvYtBeJHO#ei<=Xw4Rk@ydZM~C}bjzoTDGYm&hrxb?>w;>8qVnm}5!8qa&;renB
z79T?pLHA@VikpGcYVyI<ov#L$->1L|;8Nh9jtPt7%ZrME1AmtB9N%&llS15o;`<Au
C=<|&L

literal 0
HcmV?d00001

diff --git a/ium_s434766/my_runs/_sources/sacred-pytorch1_37f3ae3f09d3a85faa1cb43617a6d59e.py b/ium_s434766/my_runs/_sources/sacred-pytorch1_37f3ae3f09d3a85faa1cb43617a6d59e.py
new file mode 100644
index 0000000..f3d18c2
--- /dev/null
+++ b/ium_s434766/my_runs/_sources/sacred-pytorch1_37f3ae3f09d3a85faa1cb43617a6d59e.py
@@ -0,0 +1,85 @@
+import torch
+import sys
+import torch.nn.functional as F
+from torch import nn
+from sklearn.metrics import accuracy_score, mean_squared_error
+import numpy as np
+import pandas as pd
+from sacred import Experiment
+from sacred.observers import FileStorageObserver
+
+np.set_printoptions(suppress=False)
+
+ex = Experiment("ium_s434766", interactive=False, save_git_info=False)
+ex.observers.append(FileStorageObserver("ium_s434766/my_runs"))
+
+@ex.config
+def my_config():
+    num_epochs = 15
+    batch_size = 16
+    learning_rate = 0.001
+
+class LogisticRegressionModel(nn.Module):
+    def __init__(self, input_dim, output_dim):
+        super(LogisticRegressionModel, self).__init__()
+        self.linear = nn.Linear(input_dim, output_dim)
+        self.sigmoid = nn.Sigmoid()
+    def forward(self, x):
+        out = self.linear(x)
+        return self.sigmoid(out)
+
+@ex.capture
+def train(num_epochs, batch_size, learning_rate, _log):
+    data_train = pd.read_csv("data_train.csv")
+    data_test  = pd.read_csv("data_test.csv")
+    FEATURES   =  ['age','hypertension','heart_disease','ever_married', 'avg_glucose_level', 'bmi']
+
+    x_train = data_train[FEATURES].astype(np.float32)
+    y_train = data_train['stroke'].astype(np.float32)
+
+    x_test = data_test[FEATURES].astype(np.float32)
+    y_test = data_test['stroke'].astype(np.float32)
+
+    fTrain = torch.from_numpy(x_train.values)
+    tTrain = torch.from_numpy(y_train.values.reshape(2945,1))
+
+    fTest= torch.from_numpy(x_test.values)
+    tTest = torch.from_numpy(y_test.values)
+
+
+    input_dim = 6
+    output_dim = 1
+    info_params = "Batch size = " + str(batch_size) + " Epochs = " + str(num_epochs)
+    _log.info(info_params)
+    model = LogisticRegressionModel(input_dim, output_dim)
+
+    criterion = torch.nn.BCELoss(reduction='mean') 
+    optimizer = torch.optim.SGD(model.parameters(), lr = learning_rate)
+
+    for epoch in range(num_epochs):
+        # print ("Epoch #",epoch)
+        model.train()
+        optimizer.zero_grad()
+        # Forward pass
+        y_pred = model(fTrain)
+        # Compute Loss
+        loss = criterion(y_pred, tTrain)
+        # print(loss.item())
+        # Backward pass
+        loss.backward()
+        optimizer.step()
+        
+    info_loss = "Last loss = " + str(loss.item())
+    _log.info(info_loss)
+    y_pred = model(fTest)
+    # print("predicted Y value: ", y_pred.data)
+
+    torch.save(model.state_dict(), 'stroke.pth')
+
+
+@ex.automain
+def my_main(num_epochs, batch_size, learning_rate, _run):
+    train()
+
+r = ex.run()
+ex.add_artifact("stroke_model/stroke.pth")
\ No newline at end of file
diff --git a/ium_s434766/my_runs/_sources/sacred-pytorch1_e0e75cc8f994d35ec0d404b605721131.py b/ium_s434766/my_runs/_sources/sacred-pytorch1_e0e75cc8f994d35ec0d404b605721131.py
new file mode 100644
index 0000000..7d24274
--- /dev/null
+++ b/ium_s434766/my_runs/_sources/sacred-pytorch1_e0e75cc8f994d35ec0d404b605721131.py
@@ -0,0 +1,85 @@
+import torch
+import sys
+import torch.nn.functional as F
+from torch import nn
+from sklearn.metrics import accuracy_score, mean_squared_error
+import numpy as np
+import pandas as pd
+from sacred import Experiment
+from sacred.observers import FileStorageObserver
+
+np.set_printoptions(suppress=False)
+
+ex = Experiment("ium_s434766", interactive=False, save_git_info=False)
+ex.observers.append(FileStorageObserver("ium_s434766/my_runs"))
+
+@ex.config
+def my_config():
+    num_epochs = 15
+    batch_size = 16
+    learning_rate = 0.001
+
+class LogisticRegressionModel(nn.Module):
+    def __init__(self, input_dim, output_dim):
+        super(LogisticRegressionModel, self).__init__()
+        self.linear = nn.Linear(input_dim, output_dim)
+        self.sigmoid = nn.Sigmoid()
+    def forward(self, x):
+        out = self.linear(x)
+        return self.sigmoid(out)
+
+@ex.capture
+def train(num_epochs, batch_size, learning_rate, _log):
+    data_train = pd.read_csv("data_train.csv")
+    data_test  = pd.read_csv("data_test.csv")
+    FEATURES   =  ['age','hypertension','heart_disease','ever_married', 'avg_glucose_level', 'bmi']
+
+    x_train = data_train[FEATURES].astype(np.float32)
+    y_train = data_train['stroke'].astype(np.float32)
+
+    x_test = data_test[FEATURES].astype(np.float32)
+    y_test = data_test['stroke'].astype(np.float32)
+
+    fTrain = torch.from_numpy(x_train.values)
+    tTrain = torch.from_numpy(y_train.values.reshape(2945,1))
+
+    fTest= torch.from_numpy(x_test.values)
+    tTest = torch.from_numpy(y_test.values)
+
+
+    input_dim = 6
+    output_dim = 1
+    info_params = "Batch size = " + str(batch_size) + " Epochs = " + str(num_epochs)
+    _log.info(info_params)
+    model = LogisticRegressionModel(input_dim, output_dim)
+
+    criterion = torch.nn.BCELoss(reduction='mean') 
+    optimizer = torch.optim.SGD(model.parameters(), lr = learning_rate)
+
+    for epoch in range(num_epochs):
+        # print ("Epoch #",epoch)
+        model.train()
+        optimizer.zero_grad()
+        # Forward pass
+        y_pred = model(fTrain)
+        # Compute Loss
+        loss = criterion(y_pred, tTrain)
+        # print(loss.item())
+        # Backward pass
+        loss.backward()
+        optimizer.step()
+        
+    info_loss = "Last loss = " + str(loss.item())
+    _log.info(info_loss)
+    y_pred = model(fTest)
+    # print("predicted Y value: ", y_pred.data)
+
+    torch.save(model.state_dict(), 'stroke.pth')
+
+
+@ex.automain
+def my_main(num_epochs, batch_size, learning_rate, _run):
+    train()
+
+r = ex.run()
+ex.add_artifact("stroke.pth")
\ No newline at end of file
diff --git a/sacred-pytorch1.py b/sacred-pytorch1.py
index f3d18c2..7d24274 100644
--- a/sacred-pytorch1.py
+++ b/sacred-pytorch1.py
@@ -82,4 +82,4 @@ def my_main(num_epochs, batch_size, learning_rate, _run):
     train()
 
 r = ex.run()
-ex.add_artifact("stroke_model/stroke.pth")
\ No newline at end of file
+ex.add_artifact("stroke.pth")
\ No newline at end of file
diff --git a/stroke.pth b/stroke.pth
index 705a2066f553f495c9a0bc06474c28dbae8ace80..a6c4e3b885f2746a672f48835f57be55ad6d2d7c 100644
GIT binary patch
delta 414
zcmey)@s(r30Uizp1_m=@6H`kIbEAo8y&!xe3v+WbGn2_Gj6FR5nvWi5F)}cK@B}Ff
zQ3hd##G>Sk%(7Jdl*E!meKV+HHz$6EC{&<s$h|MXo1LSkLG|0@9wsGOQ3gvi%}|@&
zoJ<)eUu4psT)@PpAfn{C_w*zFeaAN}*sFc<&K{TeMSI?+H0}WzCEGsPVzL60k~lwu
zHr%MP)S}|d{5&@&b&w(IlbDq_;jW&n&1_l^asUg^qaZ8_#0anp=qs$=3GilQ0x9DF
z1rQK`ES3P$AdG5+2v8lW5l~OTjhOt9S&?zYWM&pCM#jl@EaFPSAcp_}&~k<VZ&o%C
On*|6NfdWFzARYiCCSZX8

delta 433
zcmey$@ttGB0bWi91_na|BTFMQ6AOchXT0D7rj{nAW|Nf|dw4!H^eo6>WMBZ{2~u8S
z3<3;^MadbNWvTipi6x2paP4kRJPc7tz|Ka}?t#x9yLL`NTgH~6eM`+J@BQ?3-M#>C
zc8=TIY}6(%U{aD3V^F~~4s5xblLEu!KqdvAp2AahAU)H+^(ar~VNw$3XRwFsDN8LX
z&dkqqbFv00)}Nff#50Lmi4*SV$=b}O^&s0MfIbCbF(5{OJwQ)k_f3E|BNIp+2M0to
z$aWSW6NFLC&;Y8#Fazubs2P**GAlDqnas#yWyK7%4+MmmA#{K@8#{<43NjW5K+*x;
PtZX1Q3lM_TL(~EQO8sAm