fixes

ykeuter · Nov 10, 2020 · 18ec6bb · 18ec6bb
1 parent dee5f0a
commit 18ec6bb
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+__pycache__/
+neat.log
diff --git a/examples/simple/main.py b/examples/simple/main.py
@@ -35,7 +35,7 @@ def make_net(genome, config, bs):
 
 def activate_net(net, states):
     outputs = net.activate(states).numpy()
-    return outputs[:, 0] > 0.5
+    return (outputs[:, 0] > 0.5).astype(int)
 
 
 @click.command()

diff --git a/pytorch_neat/recurrent_net.py b/pytorch_neat/recurrent_net.py
@@ -34,7 +34,7 @@ def dense_from_coo(shape, conns, dtype=torch.float64):
     if len(idxs) == 0:
         return mat
     rows, cols = np.array(idxs).transpose()
-    mat[torch.tensor(rows), torch.tensor(cols)] = torch.tensor(
+    mat[torch.tensor(rows, dtype=torch.long), torch.tensor(cols, dtype=torch.long)] = torch.tensor(
         weights, dtype=dtype)
     return mat