Sequential Inference

class predict_sequential.SequentialPredictions

Bases: object

dimensions = ['3d']

len_battery_test = 17

models_2d = {}

models_3d = {'segformer': 25, 'unet': 1, 'vnet': 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