PipelineStage#

class pyqit.core.PipelineStage(model, name=None, passthrough=False, trainable=True, input_slice=None)[source]#

Bases: object

One model in a QuantumPipeline.

Parameters:

  • model (BaseModel)

  • name (str, optional) – Defaults to the model’s class name.

  • passthrough (bool, default False) – Concatenate this stage’s input onto its output.

  • trainable (bool, default True) – frozen_backbone fit mode requires this False on every non-final stage.

  • input_slice (slice or array-like of int, optional) – Feed only these input columns to this stage.