IPSL_AID
- IPSL_AID package
- Submodules
- IPSL_AID.dataset module
stats()coarse_down_up()gaussian_filter()DataPreprocessorDataPreprocessor.const_varsDataPreprocessor.timeDataPreprocessor.yearDataPreprocessor.monthDataPreprocessor.dayDataPreprocessor.hourDataPreprocessor.year_normDataPreprocessor.doy_normDataPreprocessor.hour_normDataPreprocessor.time_batchsDataPreprocessor.eval_slicesDataPreprocessor.random_centersDataPreprocessor.center_trackerDataPreprocessor.tindex_trackerDataPreprocessor.new_epoch()DataPreprocessor.sample_time_steps_by_doy()DataPreprocessor.sample_random_time_indices()DataPreprocessor.load_dynamic_covariates()DataPreprocessor.generate_random_batch_centers()DataPreprocessor.generate_evaluation_slices()DataPreprocessor.extract_batch()DataPreprocessor.filter_batch()DataPreprocessor.normalize()DataPreprocessor.normalize_time()DataPreprocessor.__len__()DataPreprocessor.__getitem__()DataPreprocessor.__init__()DataPreprocessor.new_epoch()DataPreprocessor.sample_time_steps_by_doy()DataPreprocessor.sample_random_time_indices()DataPreprocessor.load_dynamic_covariates()DataPreprocessor.get_center_indices_from_latlon()DataPreprocessor.generate_random_batch_centers()DataPreprocessor.generate_evaluation_slices()DataPreprocessor.generate_region_slices()DataPreprocessor.extract_batch()DataPreprocessor.filter_batch()DataPreprocessor.normalize()DataPreprocessor.normalize_time()
create_dummy_netcdf()create_dummy_statistics_json()TestDataPreprocessorTestDataPreprocessor.__init__()TestDataPreprocessor.setUp()TestDataPreprocessor.test_stats_computation_without_existing_json()TestDataPreprocessor.test_stats_computation_with_existing_json()TestDataPreprocessor.test_coarse_down_up_with_torch_tensors()TestDataPreprocessor.test_coarse_down_up_with_numpy_arrays()TestDataPreprocessor.test_gaussian_filter_basic()TestDataPreprocessor.test_gaussian_filter_different_epsilon()TestDataPreprocessor.test_preprocessor_initialization_train_mode()TestDataPreprocessor.test_preprocessor_initialization_validation_mode()TestDataPreprocessor.test_preprocessor_initialization_cos_sin_normalization()TestDataPreprocessor.test_preprocessor_batch_size_validation()TestDataPreprocessor.test_new_epoch_method()TestDataPreprocessor.test_get_center_indices_from_latlon()TestDataPreprocessor.test_generate_random_batch_centers()TestDataPreprocessor.test_extract_batch()TestDataPreprocessor.test_generate_evaluation_slices()TestDataPreprocessor.test_generate_region_slices()TestDataPreprocessor.test_normalize_methods()TestDataPreprocessor.test_normalize_time_linear()TestDataPreprocessor.test_normalize_time_cos_sin()TestDataPreprocessor.test_dataset_len_method()TestDataPreprocessor.test_getitem_train_mode()TestDataPreprocessor.test_getitem_validation_mode()TestDataPreprocessor.test_getitem_with_filter_enabled()TestDataPreprocessor.test_invalid_coordinate_handling()TestDataPreprocessor.test_preprocessor_without_constants()TestDataPreprocessor.test_invalid_time_normalization()TestDataPreprocessor.tearDown()
- IPSL_AID.diagnostics module
PlotConfigPlotConfig.DEFAULT_SAVE_DIRPlotConfig.DEFAULT_FIGSIZE_MULTIPLIERPlotConfig.COLORMAPSPlotConfig.FIXED_DIFF_RANGESPlotConfig.FIXED_DIFF_RANGES_ERRORSPlotConfig.FIXED_MAE_RANGESPlotConfig.FIXED_SSR_RANGESPlotConfig.COASTLINE_wPlotConfig.BORDER_wPlotConfig.LAKE_wPlotConfig.BORDER_STYLEPlotConfig.COLORBAR_hPlotConfig.COLORBAR_PADPlotConfig.get_colormap()PlotConfig.get_plot_name()PlotConfig.convert_units()PlotConfig.get_fixed_diff_range()PlotConfig.get_fixed_diff_range_errors()PlotConfig.get_fixed_mae_range()PlotConfig.get_fixed_ssr_range()
plot_validation_hexbin()plot_comparison_hexbin()plot_metric_histories()plot_metrics_heatmap()plot_loss_histories()plot_average_metrics()plot_spatiotemporal_histograms()plot_surface()plot_ensemble_surface()plot_zoom_comparison()plot_global_surface_robinson()plot_MAE_map()plot_error_map()spread_skill_ratio()plot_spread_skill_ratio_map()plot_spread_skill_ratio_hexbin()plot_validation_pdfs()plot_power_spectra()calculate_psd2d_simple()radial_average_psd()plot_qq_quantiles()dry_frequency_map()plot_dry_frequency_map()calculate_pearsoncorr_nparray()plot_validation_mvcorr_space()plot_validation_mvcorr()plot_temporal_series_comparison()ranks()plot_ranks()get_divergence()get_curl()plot_mean_divergence_map()plot_mean_curl_map()TestPlottingFunctionsTestPlottingFunctions.__init__()TestPlottingFunctions.setUp()TestPlottingFunctions.test_validation_hexbin_comprehensive()TestPlottingFunctions.test_validation_pdfs_comprehensive()TestPlottingFunctions.test_power_spectra_comprehensive()TestPlottingFunctions.test_spatiotemporal_histograms_comprehensive()TestPlottingFunctions.test_plot_surface_comprehensive()TestPlottingFunctions.test_plot_ensemble_surface_comprehensive()TestPlottingFunctions.test_plot_zoom_comparison_comprehensive()TestPlottingFunctions.test_plot_global_surface_robinson_comprehensive()TestPlottingFunctions.test_plot_mae_map_comprehensive()TestPlottingFunctions.test_plot_error_map_comprehensive()TestPlottingFunctions.test_plot_spread_skill_ratio_map_comprehensive()TestPlottingFunctions.test_plot_spread_skill_ratio_hexbin_comprehensive()TestPlottingFunctions.test_plot_mean_divergence_map_comprehensive()TestPlottingFunctions.test_plot_mean_curl_map_comprehensive()TestPlottingFunctions.test_plot_dry_frequency_map_comprehensive()TestPlottingFunctions.test_dry_frequency_map()TestPlottingFunctions.test_divergence()TestPlottingFunctions.test_curl()TestPlottingFunctions.test_metric_plots_comprehensive()TestPlottingFunctions.test_plot_metrics_heatmap_comprehensive()TestPlottingFunctions.test_qq_quantiles_comprehensive()TestPlottingFunctions.test_mv_correlation()TestPlottingFunctions.test_temporal_series_comparison_comprehensive()TestPlottingFunctions.test_ranks()TestPlottingFunctions.test_plot_ranks()TestPlottingFunctions.tearDown()
TestSSRFunction
- IPSL_AID.evaluater module
MetricTrackermae_all()nmae_all()crps_ensemble_all()rmse_all()r2_all()pearson_all()kl_divergence_all()denormalize()edm_sampler()sampler()reconstruct_original_layout()generate_residuals_norm()run_validation()TestMetricTrackerTestMetricTracker.__init__()TestMetricTracker.setUp()TestMetricTracker.test_metric_tracker_init()TestMetricTracker.test_metric_tracker_reset()TestMetricTracker.test_metric_tracker_update()TestMetricTracker.test_metric_tracker_getmean()TestMetricTracker.test_metric_tracker_getstd()TestMetricTracker.test_metric_tracker_getsqrtmean()TestMetricTracker.test_metric_tracker_example_from_docstring()
TestErrorMetricsTestErrorMetrics.__init__()TestErrorMetrics.setUp()TestErrorMetrics.test_basic()TestErrorMetrics.test_exact_match()TestErrorMetrics.test_multi_dimensional()TestErrorMetrics.test_different_shapes()TestErrorMetrics.test_dtype_preservation()TestErrorMetrics.test_example_from_docstring()TestErrorMetrics.test_kl_divergence_basic()TestErrorMetrics.test_kl_different_shapes()TestErrorMetrics.test_kl_dtype_preservation()TestErrorMetrics.test_kl_multi_dimensional()
TestCRPSFunctionTestCRPSFunction.__init__()TestCRPSFunction.setUp()TestCRPSFunction.test_crps_basic()TestCRPSFunction.test_crps_zero_when_perfect_prediction()TestCRPSFunction.test_crps_equals_mae_for_single_member()TestCRPSFunction.test_crps_multi_dimensional_flatten()TestCRPSFunction.test_crps_dtype_preservation()
TestDenormalizeFunctionTestDenormalizeFunction.__init__()TestDenormalizeFunction.setUp()TestDenormalizeFunction.test_denormalize_minmax()TestDenormalizeFunction.test_denormalize_minmax_11()TestDenormalizeFunction.test_denormalize_standard()TestDenormalizeFunction.test_denormalize_robust()TestDenormalizeFunction.test_denormalize_log1p_minmax()TestDenormalizeFunction.test_denormalize_log1p_standard()TestDenormalizeFunction.test_denormalize_zero_denominator()TestDenormalizeFunction.test_denormalize_unsupported_type()
TestRunValidation
- IPSL_AID.logger module
LoggerTestLoggerTestLogger.__init__()TestLogger.setUp()TestLogger.test_initialization_default()TestLogger.test_initialization_with_file_output()TestLogger.test_initialization_with_record()TestLogger.test_initialization_console_only()TestLogger.test_initialization_file_only()TestLogger.test_clear_logs_with_record()TestLogger.test_clear_logs_without_record()TestLogger.test_show_header_console()TestLogger.test_show_header_file()TestLogger.test_show_header_both()TestLogger.test_start_task_minimal()TestLogger.test_start_task_with_description()TestLogger.test_start_task_with_metadata()TestLogger.test_start_task_file_output()TestLogger.test_info_console()TestLogger.test_info_file()TestLogger.test_warning_console()TestLogger.test_warning_file()TestLogger.test_success_console()TestLogger.test_success_file()TestLogger.test_step_console()TestLogger.test_step_file()TestLogger.test_error_without_exception()TestLogger.test_error_with_exception()TestLogger.test_error_file_output()TestLogger.test_exception_without_exception()TestLogger.test_exception_with_exception()TestLogger.test_exception_file_output()TestLogger.test_log_metrics_empty()TestLogger.test_log_metrics_with_data()TestLogger.test_log_metrics_file_output()TestLogger.test_full_logging_cycle()TestLogger.test_progress_bar()TestLogger.test_logger_with_unicode()TestLogger.tearDown()
- IPSL_AID.loss module
- IPSL_AID.main module
- IPSL_AID.model module
load_model_and_loss()TestModelLoaderTestModelLoader.__init__()TestModelLoader.setUp()TestModelLoader.test_ddpmpp_vp_combination()TestModelLoader.test_ncsnpp_ve_combination()TestModelLoader.test_adm_edm_combination()TestModelLoader.test_adm_unet_combination()TestModelLoader.test_rectangular_resolution()TestModelLoader.test_model_kwargs_override()TestModelLoader.test_no_conditional_channels()TestModelLoader.test_invalid_combinations()TestModelLoader.tearDown()
- IPSL_AID.model_utils module
ModelUtilsModelUtils.__init__()ModelUtils.get_parameter_number()ModelUtils.print_model_layers()ModelUtils.save_checkpoint()ModelUtils.load_checkpoint()ModelUtils.load_training_checkpoint()ModelUtils.count_parameters_by_layer()ModelUtils.log_model_summary()ModelUtils.save_training_checkpoint()ModelUtils.save_emergency_checkpoint()
TestModelTestModelUtilsTestModelUtils.__init__()TestModelUtils.setUp()TestModelUtils.test_get_parameter_number()TestModelUtils.test_get_parameter_number_with_frozen_layers()TestModelUtils.test_count_parameters_by_layer()TestModelUtils.test_print_model_layers()TestModelUtils.test_log_model_summary_without_input_shape()TestModelUtils.test_log_model_summary_with_input_shape()TestModelUtils.test_save_checkpoint()TestModelUtils.test_load_checkpoint()TestModelUtils.test_load_checkpoint_without_optimizer()TestModelUtils.test_load_training_checkpoint()TestModelUtils.test_load_training_checkpoint_nonexistent()TestModelUtils.test_save_training_checkpoint_epoch_type()TestModelUtils.test_save_training_checkpoint_best_type()TestModelUtils.test_save_training_checkpoint_final_type()TestModelUtils.test_save_training_checkpoint_samples_type()TestModelUtils.test_save_emergency_checkpoint()TestModelUtils.test_full_checkpoint_cycle()TestModelUtils.tearDown()
- IPSL_AID.networks module
weight_init()LinearConv2dGroupNormAttentionOpUNetBlockPositionalEmbeddingFourierEmbeddingSongUNetDhariwalUNetVPPrecondVEPrecondiDDPMPrecondiDDPMPrecond.img_resolutioniDDPMPrecond.in_channelsiDDPMPrecond.out_channelsiDDPMPrecond.label_dimiDDPMPrecond.use_fp16iDDPMPrecond.sigma_miniDDPMPrecond.sigma_maxiDDPMPrecond.uiDDPMPrecond.modeliDDPMPrecond.__init__()iDDPMPrecond.forward()iDDPMPrecond.alpha_bar()iDDPMPrecond.round_sigma()
EDMPrecondTestDiffusionNetworksTestDiffusionNetworks.__init__()TestDiffusionNetworks.setUp()TestDiffusionNetworks.test_song_unet_square_resolution()TestDiffusionNetworks.test_song_unet_rectangular_resolution()TestDiffusionNetworks.test_dhariwal_unet()TestDiffusionNetworks.test_vp_preconditioner()TestDiffusionNetworks.test_ve_preconditioner()TestDiffusionNetworks.test_edm_preconditioner()TestDiffusionNetworks.test_parameter_counts()TestDiffusionNetworks.tearDown()
- IPSL_AID.utils module
EasyDictFileUtilsTestEasyDictTestEasyDict.__init__()TestEasyDict.setUp()TestEasyDict.test_empty_initialization()TestEasyDict.test_initialization_with_dict()TestEasyDict.test_initialization_with_kwargs()TestEasyDict.test_attribute_get_set()TestEasyDict.test_dict_get_set()TestEasyDict.test_mixed_access()TestEasyDict.test_attribute_error_for_nonexistent()TestEasyDict.test_key_error_for_nonexistent()TestEasyDict.test_delattr()TestEasyDict.test_delitem()TestEasyDict.test_dict_methods()TestEasyDict.test_nested_easydict()TestEasyDict.test_easydict_with_easydict()TestEasyDict.tearDown()
TestFileUtilsTestFileUtils.__init__()TestFileUtils.setUp()TestFileUtils.test_makedir_new_directory()TestFileUtils.test_makedir_existing_directory()TestFileUtils.test_makedir_multiple_nested_directories()TestFileUtils.test_makefile_new_file()TestFileUtils.test_makefile_in_nonexistent_directory()TestFileUtils.test_makefile_existing_file()TestFileUtils.test_makefile_multiple_files()TestFileUtils.test_makedir_then_makefile()TestFileUtils.tearDown()
- IPSL_AID.download_ERA5_cds module
- IPSL_AID.generate_all_data_ERA5 module