created the structure for the U-net file

BrainMapperUNet.py

+"""Brain Mapper U-Net Architecture
+
+Description:
+-------------
+This folder contains the Pytorch implementation of the core U-net architecture.
+This arcitecture predicts functional connectivity rsfMRI from structural connectivity information from dMRI.
+
+Usage
+-------------
+To use this module, import it and instantiate is as you wish:
+
+    from BrainMapperUNet import BrainMapperUNet
+    deep_learning_model = BrainMapperUnet(parameters)
+
+"""
+
+import numpy as np
+import torch
+import torch.nn as nn
+
+class BrainMapperUnet(nn.Module):
+    """
+    Description
+    """
+    
+    def __init__(self, parameters):
+        pass
+
+    def forward(self, X):
+        """
+        Description
+        """
+        return None
+
+    def enable_test_dropout(self):
+        """
+        Description
+        """
+        return None
+    
+    def save(self, path):
+        """
+        Description
+        """
+        pass
+
+    def predict(self, X):
+        """
+        Description
+        """
+        return None