From 7f312dcba9dfb33bd8d92c60919192abbf6f1f6f Mon Sep 17 00:00:00 2001
From: ltcptgeneral <35508619+ltcptgeneral@users.noreply.github.com>
Date: Sat, 16 Mar 2019 13:12:59 -0500
Subject: [PATCH] Update titanlearn.py

---
 data analysis/titanlearn.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/data analysis/titanlearn.py b/data analysis/titanlearn.py
index a32bceaa..59920869 100644
--- a/data analysis/titanlearn.py	
+++ b/data analysis/titanlearn.py	
@@ -30,9 +30,10 @@ from sklearn import metrics, datasets
 import numpy as np
 import matplotlib.pyplot as plt
 import math
+import time
 
 #enable CUDA if possible
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+device = torch.device("cpu")
 
 #linear_nn: creates a fully connected network given params
 def linear_nn(in_dim, hidden_dim, out_dim, num_hidden, act_fn="tanh", end="none"):
@@ -194,8 +195,12 @@ def retyuoipufdyu():
 
     data = torch.tensor(datasets.fetch_california_housing()['data']).to(torch.float)
     ground = datasets.fetch_california_housing()['target']
-    ground=torch.tensor(ground).to(torch.float)
+    ground = torch.tensor(ground).to(torch.float)
     model = linear_nn(8, 100, 1, 20, act_fn = "relu")
     print(model)
     return train_sgd_simple(model,"regression", data, ground, learnrate=1e-4, iters=1000)
-#retyuoipufdyu()
+
+start = time.time()
+retyuoipufdyu()
+end = time.time()
+print(end-start)