pub struct BurnTabFMModel { /* private fields */ }Implementations§
Source§impl BurnTabFMModel
impl BurnTabFMModel
pub fn load(gguf_path: &Path, tc: &TabFMConfig) -> Result<Self>
pub fn is_classifier(&self) -> bool
pub fn predict( &self, x: &[Vec<f32>], y: &[f32], train_size: usize, cat_mask: Option<&[bool]>, d: Option<usize>, ) -> Result<Vec<Vec<f32>>>
pub fn predict_batch( &self, x_batch: &[Vec<Vec<f32>>], y_batch: &[Vec<f32>], train_size: usize, cat_mask_batch: &[Vec<bool>], d: Option<usize>, ) -> Result<Vec<Vec<Vec<f32>>>>
Auto Trait Implementations§
impl Freeze for BurnTabFMModel
impl RefUnwindSafe for BurnTabFMModel
impl Send for BurnTabFMModel
impl Sync for BurnTabFMModel
impl Unpin for BurnTabFMModel
impl UnsafeUnpin for BurnTabFMModel
impl UnwindSafe for BurnTabFMModel
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> 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