Commit 58ce85e0 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

adding numpy import

parent 73483fde
......@@ -37,6 +37,7 @@ from settings import Settings
import torch
import torch.utils.data as data
import numpy as np
from solver import Solver
from BrainMapperUNet import BrainMapperUNet3D
......@@ -216,6 +217,7 @@ def train(data_parameters, training_parameters, network_parameters, misc_paramet
print("K-fold Number: {} Loss: {}".format(k+1, k_fold_losses[k]))
print("K-fold Cross Validation Avearge Loss: {}".format(np.mean(k_fold_losses)))
def evaluate_score(training_parameters, network_parameters, misc_parameters, evaluation_parameters):
"""Mapping Score Evaluator
......
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