Struct TransferLearningOrchestrator

pub struct TransferLearningOrchestrator<F: Float + Debug> { /* private fields */ }

Advanced transfer learning orchestrator

Implementations

impl<F: Float + Debug + 'static + FromPrimitive + Clone + Zero + Add<Output = F> + Div<Output = F>> TransferLearningOrchestrator<F>

pub fn new( strategy: TransferStrategy, base_learning_rate: f64, init_strategy: WeightInitStrategy, ) -> Result<Self>

Create new transfer learning orchestrator

pub fn with_domain_adaptation(&mut self, method: AdaptationMethod)

Enable domain adaptation

pub fn with_model_surgery(&mut self, surgery: ModelSurgery)

Enable model surgery

pub fn load_pretrained_weights( &mut self, weights: HashMap<String, ArrayD<f32>>, ) -> Result<()>

Load pretrained weights

pub fn setup_transfer_learning( &mut self, layer_names: &[String], expected_shapes: Option<&HashMap<String, Vec<usize>>>, ) -> Result<TransferLearningSetupReport>

Setup transfer learning for a model

pub fn get_effective_learning_rate(&self, layer_name: &str) -> F

Get effective learning rate for a layer considering all factors

pub fn adapt_features( &self, layer_name: &str, features: &ArrayD<F>, ) -> Result<ArrayD<F>>

Apply domain adaptation to features

pub fn update_epoch(&mut self, epoch: usize) -> Result<()>

Update transfer learning state for new epoch