pub struct MLflowTracker { /* private fields */ }Expand description
MLflow integration implementation
Implementations§
Source§impl MLflowTracker
impl MLflowTracker
pub fn new(config: MLflowConfig) -> Self
Trait Implementations§
Source§impl ExperimentTracker for MLflowTracker
impl ExperimentTracker for MLflowTracker
Source§fn initialize(&mut self) -> Result<()>
fn initialize(&mut self) -> Result<()>
Initialize the tracker
Source§fn start_experiment(&mut self, metadata: &ExperimentMetadata) -> Result<String>
fn start_experiment(&mut self, metadata: &ExperimentMetadata) -> Result<String>
Start a new experiment/run
Source§fn log_parameter(&mut self, name: &str, value: ParameterValue) -> Result<()>
fn log_parameter(&mut self, name: &str, value: ParameterValue) -> Result<()>
Log a parameter
Source§fn log_metric(
&mut self,
name: &str,
value: f64,
step: Option<usize>,
) -> Result<()>
fn log_metric( &mut self, name: &str, value: f64, step: Option<usize>, ) -> Result<()>
Log a metric
Source§fn log_metrics(
&mut self,
metrics: HashMap<String, f64>,
step: Option<usize>,
) -> Result<()>
fn log_metrics( &mut self, metrics: HashMap<String, f64>, step: Option<usize>, ) -> Result<()>
Log multiple metrics
Source§fn log_artifact(&mut self, artifact: &ArtifactInfo) -> Result<()>
fn log_artifact(&mut self, artifact: &ArtifactInfo) -> Result<()>
Log an artifact
Source§fn log_model(
&mut self,
model_path: &PathBuf,
metadata: HashMap<String, String>,
) -> Result<()>
fn log_model( &mut self, model_path: &PathBuf, metadata: HashMap<String, String>, ) -> Result<()>
Log a model
Source§fn log_system_info(&mut self, info: HashMap<String, String>) -> Result<()>
fn log_system_info(&mut self, info: HashMap<String, String>) -> Result<()>
Log system information
Source§fn update_status(&mut self, status: ExperimentStatus) -> Result<()>
fn update_status(&mut self, status: ExperimentStatus) -> Result<()>
Update experiment status
Source§fn end_experiment(&mut self) -> Result<()>
fn end_experiment(&mut self) -> Result<()>
End the experiment
Auto Trait Implementations§
impl Freeze for MLflowTracker
impl RefUnwindSafe for MLflowTracker
impl Send for MLflowTracker
impl Sync for MLflowTracker
impl Unpin for MLflowTracker
impl UnsafeUnpin for MLflowTracker
impl UnwindSafe for MLflowTracker
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
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§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