Index _ | A | B | C | D | E | F | G | I | J | K | L | M | N | O | P | R | S | T | U | V | X _ __call__() (AcquisitionFunction method) (AcquisitionFunctionParallel method) (ContextDist method) (ConvNP method) (DataProcessor method) (ExpectedImprovement method) (GreedyAlgorithm method) (JointEntropy method) (MeanMarginalEntropy method) (MeanStddev method) (MeanVariance method) (OracleJointNLL method) (OracleMAE method) (OracleMarginalNLL method) (OracleRMSE method) (pNormStddev method) (Random method) (Stddev method) (TaskLoader method) (Trainer method) A acquisition_fn() (in module deepsensor.plot) AcquisitionFunction (class in deepsensor.active_learning.acquisition_fns) AcquisitionFunctionOracle (class in deepsensor.active_learning.acquisition_fns) AcquisitionFunctionParallel (class in deepsensor.active_learning.acquisition_fns) add_batch_dim() (Task method) add_to_config() (DataProcessor method) add_valid_time_coord_to_pred_and_move_time_dims() (in module deepsensor.model.model) alpha() (ConvNP method) append_obs_to_task() (in module deepsensor.data.task) ar_sample() (ConvNP method) assign() (Prediction method) B beta() (ConvNP method) C cast_to_float32() (Task method) check_params_computed() (DataProcessor method) compute_encoding_tensor() (in module deepsensor.model.nps) compute_greatest_data_density() (in module deepsensor.model.defaults) compute_pandas_data_resolution() (in module deepsensor.data.utils) compute_xarray_data_resolution() (in module deepsensor.data.utils) concat_tasks() (in module deepsensor.data.task) (in module deepsensor.model.convnp) config_fname (DataProcessor attribute) (TaskLoader attribute) construct_circ_time_ds() (in module deepsensor.data.utils) construct_neural_process() (in module deepsensor.model.nps) construct_x1x2_ds() (in module deepsensor.data.utils) context_encoding() (in module deepsensor.plot) ContextDist (class in deepsensor.active_learning.acquisition_fns) convert_task_to_nps_args() (in module deepsensor.model.nps) convert_to_tensor() (Task method) ConvNP (class in deepsensor.model.convnp) count_context_and_target_data_dims() (TaskLoader method) covariance() (ConvNP method) (ProbabilisticModel method) create_empty_spatiotemporal_xarray() (in module deepsensor.model.pred) D da1_da2_same_grid() (in module deepsensor.data.processor) DataProcessor (class in deepsensor.data.processor) deepsensor.active_learning.acquisition_fns module deepsensor.active_learning.algorithms module deepsensor.data.processor module deepsensor.data.sources module deepsensor.data.task module deepsensor.data.utils module deepsensor.model.convnp module deepsensor.model.defaults module deepsensor.model.model module deepsensor.model.nps module deepsensor.model.pred module deepsensor.plot module deepsensor.train.train module DeepSensorModel (class in deepsensor.model.model) E ExpectedImprovement (class in deepsensor.active_learning.acquisition_fns) extent_str_to_tuple() (in module deepsensor.plot) F feature_maps() (in module deepsensor.plot) flatten_gridded_data() (Task method) flatten_X() (in module deepsensor.data.task) flatten_Y() (in module deepsensor.data.task) G gen_decoder_scale() (in module deepsensor.model.defaults) gen_encoder_scales() (in module deepsensor.model.defaults) get_config() (DataProcessor method) get_earthenv_auxiliary_data() (in module deepsensor.data.sources) get_era5_reanalysis_data() (in module deepsensor.data.sources) get_ghcnd_station_data() (in module deepsensor.data.sources) get_gldas_land_mask() (in module deepsensor.data.sources) GreedyAlgorithm (class in deepsensor.active_learning.algorithms) I increase_spatial_resolution() (in module deepsensor.model.pred) infer_context_and_target_var_IDs() (TaskLoader method) infer_prediction_modality_from_X_t() (in module deepsensor.model.pred) interp_da1_to_da2() (in module deepsensor.data.processor) J joint_entropy() (ConvNP method) (ProbabilisticModel method) JointEntropy (class in deepsensor.active_learning.acquisition_fns) K k() (ConvNP method) L load() (ConvNP method) load_dask() (DataProcessor class method) (TaskLoader method) logpdf() (ConvNP method) (ProbabilisticModel method) loss() (ProbabilisticModel method) loss_fn() (ConvNP method) M main() (in module deepsensor.model.model) map() (DataProcessor method) map_array() (DataProcessor method) map_coord_array() (DataProcessor method) map_coords() (DataProcessor method) map_x1_and_x2() (DataProcessor method) mask_coord_array_normalised() (in module deepsensor.data.processor) mask_nans_nps() (Task method) mask_nans_numpy() (Task method) mean() (ConvNP method) (ProbabilisticModel method) mean_marginal_entropy() (ConvNP method) (ProbabilisticModel method) MeanMarginalEntropy (class in deepsensor.active_learning.acquisition_fns) MeanStddev (class in deepsensor.active_learning.acquisition_fns) MeanVariance (class in deepsensor.active_learning.acquisition_fns) min_or_max (AcquisitionFunction attribute) (ContextDist attribute) (ExpectedImprovement attribute) (JointEntropy attribute) (MeanMarginalEntropy attribute) (MeanStddev attribute) (MeanVariance attribute) (OracleJointNLL attribute) (OracleMAE attribute) (OracleMarginalNLL attribute) (OracleRMSE attribute) (pNormStddev attribute) (Random attribute) (Stddev attribute) mixture_probs() (ConvNP method) modify_task() (ConvNP class method) module deepsensor.active_learning.acquisition_fns deepsensor.active_learning.algorithms deepsensor.data.processor deepsensor.data.sources deepsensor.data.task deepsensor.data.utils deepsensor.model.convnp deepsensor.model.defaults deepsensor.model.model deepsensor.model.nps deepsensor.model.pred deepsensor.plot deepsensor.train.train N N_mixture_components (DeepSensorModel attribute) O offgrid_context() (in module deepsensor.plot) offgrid_context_observations() (in module deepsensor.plot) op() (Task method) OracleJointNLL (class in deepsensor.active_learning.acquisition_fns) OracleMAE (class in deepsensor.active_learning.acquisition_fns) OracleMarginalNLL (class in deepsensor.active_learning.acquisition_fns) OracleRMSE (class in deepsensor.active_learning.acquisition_fns) P placements() (in module deepsensor.plot) pNormStddev (class in deepsensor.active_learning.acquisition_fns) predict() (DeepSensorModel method) Prediction (class in deepsensor.model.pred) prediction() (in module deepsensor.plot) ProbabilisticModel (class in deepsensor.model.model) process_X_mask_for_X() (in module deepsensor.data.processor) R Random (class in deepsensor.active_learning.acquisition_fns) receptive_field() (in module deepsensor.plot) remove_context_nans() (Task method) remove_target_nans() (Task method) run_nps_model() (in module deepsensor.model.nps) run_nps_model_ar() (in module deepsensor.model.nps) S sample() (ConvNP method) (ProbabilisticModel method) sample_da() (TaskLoader method) sample_df() (TaskLoader method) sample_offgrid_aux() (TaskLoader method) save() (ConvNP method) (DataProcessor method) (TaskLoader method) scale() (ConvNP method) set_coord_params() (DataProcessor method) set_gpu_default_device() (in module deepsensor.train.train) slice_diag() (ConvNP method) std() (ConvNP method) (ProbabilisticModel method) Stddev (class in deepsensor.active_learning.acquisition_fns) stddev() (ProbabilisticModel method) summarise_repr() (Task class method) summarise_str() (Task class method) T Task (class in deepsensor.data.task) task() (in module deepsensor.plot) task_generation() (TaskLoader method) TaskLoader (class in deepsensor.data.loader) time_slice_variable() (TaskLoader method) train_epoch() (in module deepsensor.train.train) Trainer (class in deepsensor.train.train) U unnormalise() (DataProcessor method) V variance() (ConvNP method) (ProbabilisticModel method) X xarray_to_coord_array_normalised() (in module deepsensor.data.processor)
