Training sequential Networks

class train_sequential.SequentialTrain

Bases: object

dataset = 'btcv'

dimensions = ['3d']

len_battery_test = 17

models_2d = {}

models_3d = {'attention_unet': 1}

static run_model(model, dimension, run_version)

Function to run the model to predict the battery test. When the prediction is finished, the time with more parameters are saved in a json file. The place for the json file is in the respective folder results.

Parameters:

• model (str) – Name of the model

• dimension (str) – Number of dimensions (2d or 3d)

• run_version (str) – Version of the model