IPSL-AID API
- IPSL_AID package
- Submodules
- IPSL_AID.dataset module
stats()coarse_down_up()gaussian_filter()make_starts()DataPreprocessorDataPreprocessor.const_varsDataPreprocessor.timeDataPreprocessor.yearDataPreprocessor.monthDataPreprocessor.dayDataPreprocessor.hourDataPreprocessor.year_normDataPreprocessor.doy_normDataPreprocessor.hour_normDataPreprocessor.time_batchsDataPreprocessor.eval_slicesDataPreprocessor.eval_region_indicesDataPreprocessor.train_slicesDataPreprocessor.region_centerDataPreprocessor.region_sizeDataPreprocessor.overlap_ratioDataPreprocessor.random_centersDataPreprocessor.center_trackerDataPreprocessor.tindex_trackerDataPreprocessor.__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.valid_multiregional_centers()DataPreprocessor.generate_random_batch_centers()DataPreprocessor.generate_evaluation_slices()DataPreprocessor.generate_region_slices()DataPreprocessor.extract_batch()DataPreprocessor.build_fine_coarse_blocks()DataPreprocessor.load_features()DataPreprocessor.filter_batch()DataPreprocessor.normalize()DataPreprocessor.normalize_time()
- 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()compute_sal()compute_sal_objects()compute_sal_objects_time()plot_validation_salpdfs()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()
- IPSL_AID.evaluater module
- IPSL_AID.logger module
- IPSL_AID.loss module
- IPSL_AID.main module
- IPSL_AID.model module
- 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()
- 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()
EDMPrecond
- IPSL_AID.utils module
- IPSL_AID.download_ERA5_cds module
- IPSL_AID.generate_all_data_ERA5 module
- AID_BC package
- Submodules
- AID_BC.bias_corrector module
BiasCorrectorQuantileMappingCorrectorOptimalTransportCorrectorOptimalTransportCorrector.centerOptimalTransportCorrector.scaleOptimalTransportCorrector.target_pointsOptimalTransportCorrector.target_potentialOptimalTransportCorrector.solverOptimalTransportCorrector.transport_functionOptimalTransportCorrector.convergedOptimalTransportCorrector.fitted_iterationsOptimalTransportCorrector.regularized_ot_costOptimalTransportCorrector.method_nameOptimalTransportCorrector.__init__()OptimalTransportCorrector.fit()OptimalTransportCorrector.transform()OptimalTransportCorrector.diagnostics
create_bias_corrector()
- AID_BC.dataset module
resize_with_torchvision()ClimateDatasetClimateDataset.era5_dataClimateDataset.cmip6_dataClimateDataset.__init__()ClimateDataset.load_cmip6()ClimateDataset.load_era5()ClimateDataset.rename_coordinates()ClimateDataset.is_latitude_descending()ClimateDataset.normalize_longitudes()ClimateDataset.sort_latitude()ClimateDataset.standardize_coordinates()ClimateDataset.prepare_dataset()ClimateDataset.close()
- AID_BC.diagnostics module
PlotConfigPlotConfig.DEFAULT_SAVE_DIRPlotConfig.DEFAULT_FIGSIZE_MULTIPLIERPlotConfig.COLORMAPSPlotConfig.FIXED_DIFF_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()
to_numpy_4d()plot_validation_pdfs()plot_power_spectra()calculate_psd2d_simple()radial_average_psd()plot_qq_quantiles()plot_metrics_heatmap()plot_surface()
- AID_BC.evaluater module
- AID_BC.logger module
- AID_BC.main module
parse_args()parse_variable_paths()validate_variable_paths()validate_args()build_path()check_same_spatial_grid()check_dataset_consistency()select_month()prepare_dataarray()resolve_precomputed_variable_root()load_precomputed_metadata()open_dataarray()load_training_variable()assemble_variable_dataset()load_training_data()load_application_data()flatten_dataset()reconstruct_dataset()fit_bias_corrector()fit_bias_correctors_by_month()apply_bias_corrector()apply_bias_corrector_by_month()save_corrected_year()main()
- AID_BC.optimal_transport module
TransportSolutionTransportSolution.potentialsTransportSolution.cost_matrixTransportSolution.epsilonTransportSolution.reg_ot_costTransportSolution.thresholdTransportSolution.convergedTransportSolution.num_iterationsTransportSolution.fuTransportSolution.gvTransportSolution.costTransportSolution.transport_plan
OptimalTransport
- AID_BC.preprocess module
- AID_BC.quantile_mapping module
- AID_BC.version module