pub struct TabDptConfig {Show 13 fields
pub dim: usize,
pub n_layers: usize,
pub n_heads: usize,
pub ff_dim: usize,
pub y_encoder_dim: usize,
pub max_num_classes: usize,
pub regression_bin_count: usize,
pub regression_bin_min: f32,
pub regression_bin_max: f32,
pub max_num_features: usize,
pub base_len: usize,
pub max_len: usize,
pub n_thinking_rows: usize,
}Expand description
TabDPT model configuration. One checkpoint (Layer6/TabDPT) serves both classification and
regression — the head produces max_num_classes + regression_bin_count outputs; callers
slice whichever half they need.
Fields§
§dim: usize§n_layers: usize§n_heads: usize§ff_dim: usize§y_encoder_dim: usize§max_num_classes: usize§regression_bin_count: usize§regression_bin_min: f32§regression_bin_max: f32§max_num_features: usize§base_len: usize§max_len: usize§n_thinking_rows: usizeImplementations§
Source§impl TabDptConfig
impl TabDptConfig
Sourcepub fn default_v1_2() -> Self
pub fn default_v1_2() -> Self
Matches Layer6/TabDPT’s tabdpt1_2.safetensors embedded config exactly.
Trait Implementations§
Source§impl Clone for TabDptConfig
impl Clone for TabDptConfig
Source§fn clone(&self) -> TabDptConfig
fn clone(&self) -> TabDptConfig
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 TabDptConfig
impl RefUnwindSafe for TabDptConfig
impl Send for TabDptConfig
impl Sync for TabDptConfig
impl Unpin for TabDptConfig
impl UnsafeUnpin for TabDptConfig
impl UnwindSafe for TabDptConfig
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