Skip to main content

Module builder

Module builder 

Source
Expand description

Fluent builder for common Deep Kernel topologies.

Mirrors the style of crate::learned_composition::LearnedMixtureBuilder: chained setters plus a terminating DeepKernelBuilder::build that returns Result<DeepKernel<MLPFeatureExtractor, K>>.

Typical usage:

use tensorlogic_sklears_kernels::{
    deep_kernel::{Activation, DeepKernelBuilder},
    RbfKernel, RbfKernelConfig,
};

let rbf = RbfKernel::new(RbfKernelConfig::new(0.5)).expect("valid");
let dkl = DeepKernelBuilder::new()
    .input_dim(4)
    .hidden_layer(8, Activation::ReLU)
    .hidden_layer(4, Activation::Tanh)
    .output_dim(2, Activation::Identity)
    .seed(42)
    .build(rbf)
    .expect("valid topology");
let _ = dkl;

Structsยง

DeepKernelBuilder
Fluent builder for Deep Kernel networks.