Predictor¶

The main module for run inference

class neural_pipeline.predict.Predictor(model: neural_pipeline.data_processor.model.Model, fsm: neural_pipeline.utils.fsm.FileStructManager, from_best_state: bool = False)[source]¶

Predictor run inference by training parameters

Parameters:	model – model object, used for predict fsm – `FileStructManager` object

predict(data: torch.Tensor)[source]¶

Predict ine data

Parameters:	data – data as `torch.Tensor` or dict with key `data`
Returns:	processed output
Return type:	model output type