Commit 2f498e71 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

created the structure for the U-net file

parent 9c4903d5
"""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
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment