Struct PretrainedModel 

pub struct PretrainedModel {
    pub model: Box<dyn PipelinePredictor>,
    pub frozen_layers: Vec<String>,
    pub trainable_layers: Vec<String>,
    pub metadata: HashMap<String, String>,
}

Pre-trained model wrapper for transfer learning

Fields

model: Box<dyn PipelinePredictor>

The pre-trained model

frozen_layers: Vec<String>

Feature extraction layers (frozen)

trainable_layers: Vec<String>

Fine-tuning layers (trainable)

metadata: HashMap<String, String>

Model metadata

Implementations

impl PretrainedModel

pub fn new(model: Box<dyn PipelinePredictor>) -> Self

Create a new pre-trained model wrapper

pub fn with_frozen_layers(self, layers: Vec<String>) -> Self

Add frozen layers

pub fn with_trainable_layers(self, layers: Vec<String>) -> Self

Add trainable layers

pub fn with_metadata(self, metadata: HashMap<String, String>) -> Self

Add metadata

pub fn extract_features( &self, x: &ArrayView2<'_, Float>, ) -> SklResult<Array2<f64>>

Extract features using frozen layers

Trait Implementations

impl Debug for PretrainedModel

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.