Fix neural network

1 files changed, 3 insertions, 5 deletions

diff --git a/utils.py b/utils.py
index d068023..a4f6cd3 100644
--- a/utils.py
+++ b/utils.py
@@ -1,6 +1,4 @@
-import numpy
-from random import random
+import numpy as np
 
-def random_array(rows: int, columns: int):
-    random_matrix = [[random() for x in range(rows)] for y in range(columns)]
-    return numpy.asarray(random_matrix)
+def random_array(length: int):
+    return np.random.rand(length, 1)