cybench package

Subpackages

• cybench.datasets package
  • Submodules
  • cybench.datasets.alignment module
    • _add_cutoff_days()
    • align_data()
    • trim_to_lead_time()
  • cybench.datasets.configured module
    • _add_year()
    • _preprocess_time_series_data()
    • load_dfs()
    • load_dfs_crop()
  • cybench.datasets.dataset module
    • Dataset
      • Dataset._empty_df_target()
      • Dataset._filter_df_on_index()
      • Dataset._get_feature_data()
      • Dataset._split_df_on_index()
      • Dataset._validate_dfs()
      • Dataset.crop
      • Dataset.feature_names
      • Dataset.indices()
      • Dataset.load()
      • Dataset.location_ids
      • Dataset.max_date
      • Dataset.min_date
      • Dataset.split_on_years()
      • Dataset.targets()
      • Dataset.years
  • cybench.datasets.dataset_overview module
  • cybench.datasets.dataset_torch module
    • TorchDataset
      • TorchDataset._cast_to_tensor()
      • TorchDataset.collate_fn()
  • cybench.datasets.transforms module
    • _transform_ts_input_to_dekadal()
    • transform_stack_ts_static_inputs()
    • transform_ts_inputs_to_dekadal()
  • Module contents
• cybench.evaluation package
  • Submodules
  • cybench.evaluation.eval module
    • evaluate_model()
    • evaluate_predictions()
    • get_default_metrics()
    • mape()
    • metric()
    • normalized_rmse()
  • cybench.evaluation.example_log_experiment module
  • cybench.evaluation.log_experiments module
  • Module contents
• cybench.models package
  • Submodules
  • cybench.models.model module
    • BaseModel
      • BaseModel._abc_impl
      • BaseModel.fit()
      • BaseModel.load()
      • BaseModel.predict()
      • BaseModel.predict_batch()
      • BaseModel.predict_item()
      • BaseModel.save()
  • cybench.models.naive_models module
    • AverageYieldModel
      • AverageYieldModel._abc_impl
      • AverageYieldModel.fit()
      • AverageYieldModel.load()
      • AverageYieldModel.predict_batch()
      • AverageYieldModel.save()
  • cybench.models.nn_models module
    • BaseNNModel
      • BaseNNModel._abc_impl
      • BaseNNModel.fit()
      • BaseNNModel.load()
      • BaseNNModel.predict_batch()
      • BaseNNModel.save()
      • BaseNNModel.train_model()
    • ExampleLSTM
      • ExampleLSTM._abc_impl
      • ExampleLSTM.forward()
  • cybench.models.sklearn_model module
    • SklearnModel
      • SklearnModel._abc_impl
      • SklearnModel._design_features()
      • SklearnModel._optimize_hyperparameters()
      • SklearnModel.fit()
      • SklearnModel.load()
      • SklearnModel.predict()
      • SklearnModel.predict_batch()
      • SklearnModel.save()
  • cybench.models.trend_model module
    • TrendModel
      • TrendModel._abc_impl
      • TrendModel._linear_trend_estimator()
      • TrendModel._quadratic_trend_estimator()
      • TrendModel.fit()
      • TrendModel.load()
      • TrendModel.predict_batch()
      • TrendModel.save()
  • Module contents
• cybench.runs package
  • Submodules
  • cybench.runs.results_plots module
    • box_plots_metrics()
    • box_plots_residuals()
    • plot_bars()
    • plot_graph()
    • plot_metrics()
    • plot_yearly_metrics()
    • plot_yearly_residuals()
    • write_to_markdown()
  • cybench.runs.run_benchmark module
    • compute_metrics()
    • get_prediction_residuals()
    • load_results()
    • run_benchmark()
    • run_benchmark_on_all_data()
  • cybench.runs.validate_model module
    • validate_single_model()
  • Module contents
• cybench.util package
  • Submodules
  • cybench.util.data module
    • data_to_pandas()
    • flatten_nested_dict()
    • generate_settings()
    • unflatten_nested_dict()
    • update_settings()
  • cybench.util.features module
    • _add_period()
    • _aggregate_by_period()
    • _count_threshold()
    • dekad_from_date()
    • design_features()
    • fortnight_from_date()
    • growing_degree_days()
    • unpack_time_series()
  • cybench.util.torch module
    • batch_tensors()
  • Module contents

Submodules

cybench.config module

Module contents