Fixing some bugs

chrislemke · Nov 9, 2022 · 61a72f8 · 61a72f8
1 parent b4aabac
commit 61a72f8
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 10 deletions.
diff --git a/autoembedder/evaluator.py b/autoembedder/evaluator.py
@@ -86,14 +86,12 @@ def __predict(
         float: Loss value.
     """
 
-    device = (
-        torch.device(
-            "cuda"
-            if torch.cuda.is_available()
-            else "mps"
-            if torch.backends.mps.is_available() and parameters.get("use_mps", 0) == 1
-            else "cpu"
-        ),
+    device = torch.device(
+        "cuda"
+        if torch.cuda.is_available()
+        else "mps"
+        if torch.backends.mps.is_available() and parameters.get("use_mps", 0) == 1
+        else "cpu"
     )
 
     with torch.no_grad():

diff --git a/example.ipynb b/example.ipynb
@@ -282,7 +282,7 @@
    "source": [
     "parameters = {\n",
     "    \"hidden_layers\": [[25, 20], [20, 10]],\n",
-    "    \"epochs\": 1,\n",
+    "    \"epochs\": 10,\n",
     "    \"lr\": 0.0001,\n",
     "    \"verbose\": 1,\n",
     "    \"target\": \"Class\",\n",

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "Autoembedder"
-version = "0.1.13"
+version = "0.1.14"
 description = "PyTorch autoencoder with additional embeddings layer for categorical data."
 authors = ["Christopher Lemke <chris@syhbl.mozmail.com>"]
 license = "MIT"