add todos

SiddeshSambasivam · Jun 27, 2021 · 777513b · 777513b
1 parent 7acfa1e
commit 777513b
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/matterix/loss_fn.py b/matterix/loss_fn.py
@@ -0,0 +1 @@
+# TODO: MSE, RMSE, MAE, Binary cross-entropy, Categorical cross-entropy, kullback leibler divergence loss
diff --git a/matterix/nn.py b/matterix/nn.py
@@ -1,3 +1,9 @@
 # TODO: Module class to represent neural networks
-# TODO: Loss functions
-# TODO: Optimizers
+
+
+class Module:
+    def __init__(self) -> None:
+        pass
+
+    def zero_grad(self) -> None:
+        pass
diff --git a/matterix/tensor.py b/matterix/tensor.py
@@ -4,7 +4,7 @@
 ArrayableType = Union[float, list, np.ndarray]
 TensorableType = Union[float, np.ndarray, "Tensor"]
 
-# TODO: randn, normal, randint
+# TODO: randn, normal, randint, argmax
 
 
 def enforceTensor(_input: TensorableType) -> "Tensor":
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		# TODO: MSE, RMSE, MAE, Binary cross-entropy, Categorical cross-entropy, kullback leibler divergence loss