Struct SurvivalTransformerConfig 

pub struct SurvivalTransformerConfig {
    pub d_model: usize,
    pub n_heads: usize,
    pub n_layers: usize,
    pub d_ff: usize,
    pub dropout_rate: f64,
    pub max_time: f64,
    pub num_time_bins: usize,
    pub learning_rate: f64,
    pub batch_size: usize,
    pub n_epochs: usize,
    pub seed: Option<u64>,
}

Fields

d_model: usize

n_heads: usize

n_layers: usize

d_ff: usize

dropout_rate: f64

max_time: f64

num_time_bins: usize

learning_rate: f64

batch_size: usize

n_epochs: usize

seed: Option<u64>

Implementations

impl SurvivalTransformerConfig

pub fn new( d_model: usize, n_heads: usize, n_layers: usize, d_ff: usize, dropout_rate: f64, max_time: f64, num_time_bins: usize, learning_rate: f64, batch_size: usize, n_epochs: usize, seed: Option<u64>, ) -> PyResult<Self>

Trait Implementations

impl Clone for SurvivalTransformerConfig

fn clone(&self) -> SurvivalTransformerConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for SurvivalTransformerConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'py> IntoPyObject<'py> for SurvivalTransformerConfig

type Target = SurvivalTransformerConfig

The Python output type

type Output = Bound<'py, <SurvivalTransformerConfig as IntoPyObject<'py>>::Target>

The smart pointer type to use.

type Error = PyErr

The type returned in the event of a conversion error.

fn into_pyobject( self, py: Python<'py>, ) -> Result<<Self as IntoPyObject<'_>>::Output, <Self as IntoPyObject<'_>>::Error>

Performs the conversion.

impl PyClass for SurvivalTransformerConfig

type Frozen = False

Whether the pyclass is frozen.

impl PyClassImpl for SurvivalTransformerConfig

const IS_BASETYPE: bool = false

#[pyclass(subclass)]

const IS_SUBCLASS: bool = false

#[pyclass(extends=…)]

const IS_MAPPING: bool = false

#[pyclass(mapping)]

const IS_SEQUENCE: bool = false

#[pyclass(sequence)]

const IS_IMMUTABLE_TYPE: bool = false

#[pyclass(immutable_type)]

const RAW_DOC: &'static CStr = c"\x00"

Docstring for the class provided on the struct or enum.

const DOC: &'static CStr

Fully rendered class doc, including the text_signature if a constructor is defined.

type BaseType = PyAny

Base class

type ThreadChecker = SendablePyClass<SurvivalTransformerConfig>

This handles following two situations:

type PyClassMutability = <<PyAny as PyClassBaseType>::PyClassMutability as PyClassMutability>::MutableChild

Immutable or mutable

type Dict = PyClassDummySlot

Specify this class has #[pyclass(dict)] or not.

type WeakRef = PyClassDummySlot

Specify this class has #[pyclass(weakref)] or not.

type BaseNativeType = PyAny

The closest native ancestor. This is PyAny by default, and when you declare #[pyclass(extends=PyDict)], it’s PyDict.

fn items_iter() -> PyClassItemsIter

fn lazy_type_object() -> &'static LazyTypeObject<Self>

fn dict_offset() -> Option<isize>

fn weaklist_offset() -> Option<isize>

impl PyClassNewTextSignature for SurvivalTransformerConfig

const TEXT_SIGNATURE: &'static str = "(d_model=64, n_heads=4, n_layers=2, d_ff=128, dropout_rate=0.1, max_time=100.0, num_time_bins=20, learning_rate=0.001, batch_size=64, n_epochs=100, seed=None)"

impl PyMethods<SurvivalTransformerConfig> for PyClassImplCollector<SurvivalTransformerConfig>

fn py_methods(self) -> &'static PyClassItems

impl PyTypeInfo for SurvivalTransformerConfig

const NAME: &'static str = "SurvivalTransformerConfig"

Class name.

const MODULE: Option<&'static str> = ::core::option::Option::None

Module name, if any.

fn type_object_raw(py: Python<'_>) -> *mut PyTypeObject

Returns the PyTypeObject instance for this type.

fn type_object(py: Python<'_>) -> Bound<'_, PyType>

Returns the safe abstraction over the type object.

fn is_type_of(object: &Bound<'_, PyAny>) -> bool

Checks if object is an instance of this type or a subclass of this type.

fn is_exact_type_of(object: &Bound<'_, PyAny>) -> bool

Checks if object is an instance of this type.

impl DerefToPyAny for SurvivalTransformerConfig

impl ExtractPyClassWithClone for SurvivalTransformerConfig