pub struct ModelVisualizer { /* private fields */ }Expand description
Model architecture visualizer
Implementations§
Source§impl ModelVisualizer
impl ModelVisualizer
Sourcepub fn new(config: VisualizationConfig) -> Self
pub fn new(config: VisualizationConfig) -> Self
Create a new model visualizer
Sourcepub fn visualize_architecture(
&self,
layers: &[LayerInfo],
filename: &str,
) -> NeuralResult<()>
pub fn visualize_architecture( &self, layers: &[LayerInfo], filename: &str, ) -> NeuralResult<()>
Generate a model architecture diagram
Auto Trait Implementations§
impl Freeze for ModelVisualizer
impl RefUnwindSafe for ModelVisualizer
impl Send for ModelVisualizer
impl Sync for ModelVisualizer
impl Unpin for ModelVisualizer
impl UnsafeUnpin for ModelVisualizer
impl UnwindSafe for ModelVisualizer
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