Predictor

The main module for run inference

class neural_pipeline.predict.Predictor(model: neural_pipeline.data_processor.model.Model, fsm: neural_pipeline.utils.file_structure_manager.FileStructManager, device: torch.device = None)

Predictor run inference by training parameters

Parameters:

• model – model object, used for predict
• fsm – FileStructManager object
• device – device for run inference

predict(data: torch.Tensor)

Predict ine data

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

predict_dataset(data_producer: neural_pipeline.data_producer.data_producer.DataProducer, callback: callable) → None

Run prediction iterates by data_producer

Parameters:

• data_producer – DataProducer object
• callback – callback, that call for every data prediction and get it’s result as parameter