AIToolbox
latest

Components:

  • torchtrain
    • Train Loop
    • TTModel
    • Callbacks
    • Schedulers
    • Multi-Loss and Multi-Optimizer
    • Multi-GPU Training
    • Automatic Mixed Precision Training
    • Advanced Topics
      • Message Passing Service
      • Model Prediction Store
      • Model Wrap and Batch Feed Definition
  • experiment
    • Result Package
    • Result Metric
    • Experiment Saving
    • Training History
  • cloud
    • Saving to Cloud
    • Loading Models from Cloud
    • Data Access
    • AWS Simple Email Service
  • nlp

Examples:

  • Examples

API:

  • aitoolbox
    • torchtrain
      • callbacks
        • abstract
        • basic
        • ddp
        • gradient
        • model_load
        • model_save
        • performance_eval
        • tensorboard
        • train_schedule
        • wandb
      • data
        • batch_model_feed_defs
        • dataset
      • schedulers
        • basic
        • warmup
      • train_loop
        • components
          • callback_handler
          • ddp_handler
          • message_passing
          • model_prediction_store
          • pred_collate_fns
        • train_loop
        • train_loop_tracking
      • model
      • model_predict
      • multi_loss_optim
      • parallel
    • experiment
      • core_metrics
        • abstract_metric
        • classification
        • regression
      • local_load
        • local_model_load
      • local_save
        • folder_create
        • local_model_save
        • local_results_save
      • result_package
        • abstract_result_packages
        • basic_packages
        • hf_evaluate_packages
        • torch_metrics_packages
      • result_reporting
        • hyperparam_reporter
        • report_generator
      • experiment_saver
      • local_experiment_saver
      • training_history
    • cloud
      • AWS
        • data_access
        • model_load
        • model_save
        • results_save
        • simple_email_service
      • GoogleCloud
        • data_access
        • model_load
        • model_save
        • results_save
    • nlp
      • core
        • core
        • vocabulary
      • experiment_evaluation
        • NLP_metrics
        • NLP_result_package
        • attention_heatmap
      • torch_collate_fns
    • utils
      • dict_util
      • file_system
      • util
AIToolbox
  • aitoolbox
  • experiment
  • core_metrics
  • Edit on GitHub

core_metrics

Submodules

  • abstract_metric
    • AbstractBaseMetric
      • AbstractBaseMetric.calculate_metric()
      • AbstractBaseMetric.get_metric()
      • AbstractBaseMetric.get_metric_dict()
      • AbstractBaseMetric._get_metric_self_other_val()
      • AbstractBaseMetric.__add__()
      • AbstractBaseMetric.__radd__()
      • AbstractBaseMetric.concat_metric()
  • classification
    • AccuracyMetric
      • AccuracyMetric.calculate_metric()
    • ROCAUCMetric
      • ROCAUCMetric.calculate_metric()
    • PrecisionRecallCurveAUCMetric
      • PrecisionRecallCurveAUCMetric.calculate_metric()
    • F1ScoreMetric
      • F1ScoreMetric.calculate_metric()
    • PrecisionMetric
      • PrecisionMetric.calculate_metric()
    • RecallMetric
      • RecallMetric.calculate_metric()
  • regression
    • MeanSquaredErrorMetric
      • MeanSquaredErrorMetric.calculate_metric()
    • MeanAbsoluteErrorMetric
      • MeanAbsoluteErrorMetric.calculate_metric()
Previous Next

© Copyright 2023, Marko Vidoni. Revision 6b9bcfda.

Built with Sphinx using a theme provided by Read the Docs.