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ยง
- Deep
Kernel Builder - Fluent builder for Deep Kernel networks.