pub struct TabPfnConfig {Show 18 fields
pub embed_dim: usize,
pub dist_embed_num_blocks: usize,
pub dist_embed_num_heads: usize,
pub dist_embed_num_inducing_points: usize,
pub feature_group_size: usize,
pub feat_agg_num_blocks: usize,
pub feat_agg_num_heads: usize,
pub feat_agg_num_cls_tokens: usize,
pub nlayers: usize,
pub icl_num_heads: usize,
pub icl_num_kv_heads_test: Option<usize>,
pub decoder_head_dim: usize,
pub decoder_num_heads: usize,
pub decoder_use_softmax_scaling: bool,
pub ff_factor: usize,
pub softmax_scaling_mlp_hidden_dim: usize,
pub max_num_classes: usize,
pub use_nan_indicators: bool,
}Expand description
TabPFN-3 (Prior-Labs/tabpfn_3) architecture config — matches the real checkpoint’s embedded
config dict exactly (classifier and regressor share this shape; only task_type/heads
differ, and only classification is implemented here).
Fields§
§embed_dim: usize§dist_embed_num_blocks: usize§dist_embed_num_heads: usize§dist_embed_num_inducing_points: usize§feature_group_size: usize§feat_agg_num_blocks: usize§feat_agg_num_heads: usize§feat_agg_num_cls_tokens: usize§nlayers: usize§icl_num_heads: usize§icl_num_kv_heads_test: Option<usize>§decoder_head_dim: usize§decoder_num_heads: usize§decoder_use_softmax_scaling: bool§ff_factor: usize§max_num_classes: usize§use_nan_indicators: boolImplementations§
Source§impl TabPfnConfig
impl TabPfnConfig
Sourcepub fn v3_default() -> Self
pub fn v3_default() -> Self
Values from the real tabpfn-v3-classifier-v3_default.ckpt’s embedded config.
pub fn icl_dim(&self) -> usize
Sourcepub fn cell_in_features(&self) -> usize
pub fn cell_in_features(&self) -> usize
x_embed’s input width: grouped raw values, doubled if NaN indicators are concatenated.
Trait Implementations§
Source§impl Clone for TabPfnConfig
impl Clone for TabPfnConfig
Source§fn clone(&self) -> TabPfnConfig
fn clone(&self) -> TabPfnConfig
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for TabPfnConfig
impl RefUnwindSafe for TabPfnConfig
impl Send for TabPfnConfig
impl Sync for TabPfnConfig
impl Unpin for TabPfnConfig
impl UnsafeUnpin for TabPfnConfig
impl UnwindSafe for TabPfnConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> ErasedDestructor for Twhere
T: 'static,
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more