Skip to content

Commit

Permalink
more docs
Browse files Browse the repository at this point in the history
  • Loading branch information
nocollier committed Mar 5, 2024
1 parent 7554ed1 commit 6b674e8
Show file tree
Hide file tree
Showing 3 changed files with 66 additions and 8 deletions.
2 changes: 1 addition & 1 deletion doc/configure.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,4 +38,4 @@ The location to which we will download ESGF holdings is set to `${HOME}/.esgf` b

## Using data directly

This package is designed to download data only if it is absolutely necesary. If you are working on a resource with direct access to some portion of the ESGF holdings, you can point to it with `cat.set_esgf_data_root()`. This will add a read-only location to check for data. We check for a few locations automatically when the package is instantiated. If you would like a location added to our [defaults](https://github.com/esgf2-us/intake-esgf/blob/76fdb8e943f73813160bd76544d5d471c25f2a2d/intake_esgf/base.py#L169), please feel free to submit a [pull request](https://github.com/esgf2-us/intake-esgf/issues/new?assignees=&labels=&projects=&template=feature_request.md&title=).
This package is designed to download data only if it is absolutely necesary. If you are working on a resource with direct access to some portion of the ESGF holdings, you can point to it with `cat.set_esgf_data_root()`. This will add a read-only location to check for data. We check for a few locations automatically when the package is instantiated. If you would like a location added to our [defaults](https://github.com/esgf2-us/intake-esgf/blob/76fdb8e943f73813160bd76544d5d471c25f2a2d/intake_esgf/base.py#L169), please feel free to submit a [issue](https://github.com/esgf2-us/intake-esgf/issues/new?assignees=&labels=&projects=&template=feature_request.md&title=).
7 changes: 3 additions & 4 deletions doc/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -50,13 +50,12 @@ or through ``conda-forge``
:hidden:
:caption: Features

configure
measures
dictkeys
logging
modelgroups
operators
configure
reproduce
dictkeys
logging

.. toctree::
:maxdepth: 2
Expand Down
65 changes: 62 additions & 3 deletions doc/modelgroups.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,13 +9,72 @@ kernelspec:

# Simplifying Search with Model Groups

At a simple level, you can think of `intake-esgf` as analagous to the ESGF web [interface](https://aims2.llnl.gov/search) but where results are presented to you as a pandas dataframe in place of pages of web results. However, we believe that the user does not want to wade through either of these. Many times you want to see model results organized by unique combinations of `source_id`, `member_id`, and `grid_label`. That is to say, when you are going to perform an analysis, you would like your model outputs to be self-consistent and from the same run and grid, even across experiments. To assist you in honing in on what sets of results may be useful to your analysis, we introduce the notion of *model groups*.

Consider the following search, motivated by a desire to study controls (temperature, precipitation, soil moisture) on the carbon cycle (gross primary productivty) across a number of historical and future scenarios.

```{code-cell}
from intake_esgf import ESGFCatalog
cat = ESGFCatalog().search(
experiment_id="historical",
variable_id=["tas", "gpp"],
frequency="mon",
experiment_id=["historical", "ssp585", "ssp370", "ssp245", "ssp126"],
variable_id=["gpp", "tas", "pr", "mrso"],
table_id=["Amon", "Lmon"],
)
print(cat)
```

Even if this exact application does not resonate with you, the situation is a familiar one. We have several thousand results with many different models and variants to sort through. To help guide you to which groups of models might be useful to you, we provide the following function.

```{code-cell}
cat.model_groups()
```

This returns a pandas series where the results have been grouped and sorted by `source_id`, `member_id`, and `grid_label` and the counts of datasets returned. Pandas will probably truncate this series. If you want to see the whole series, you can call `print(cat.model_groups().to_string())` instead. However, as there are still several hundred possibile model groups, we will not show that here.

## Removing Incomplete Groups

If you glance through the model groups, you will see that, relative to our search, many will be *incomplete*. By this we mean, that there are many model groups that will not have all the variables in all the experiments that we wish to include in our analysis. Since we are looking for 5 experiments and 4 variables, we need the model groups with 20 dataset results. We can check which groups satisfy this condition by operating on the model group pandas series.

```{code-cell}
mgs = cat.model_groups()
print(mgs[mgs==20])
```

The rest are incomplete and we would like a fast way to remove them from the search results. But the reality is that many times our *completeness* criteria is more complicated than just a number. In the above example, we may want all the variables for all the experiments, but if a model does not have a submission for, say, `ssp126`, that is acceptable.

`intake-esgf` provides an interface which uses a user-provided function to remove incomplete entries. Internally, we will loop over all model groups in the results and pass your function the portion of the dataframe that corresponds to the current model group. Your function then needs to return a boolean based on the contents of that sub-dataframe.

```{code-cell}
def should_i_keep_it(sub_df):
# this model group has all experiments/variables
if len(sub_df) == 20:
return True
# if any of these experiments is missing a variable, remove this
for exp in ["historical", "ssp585", "ssp370", "ssp245"]:
if len(sub_df[sub_df["experiment_id"] == exp]) != 4:
return False
# if the check makes it here, keep it
return True
```

Then we pass this function to the catalog by the `remove_incomplete()` function and observe how it has impacted the search results.

```{code-cell}
cat.remove_incomplete(should_i_keep_it)
print(cat.model_groups())
```

## Removing Ensembles

Depending on the goals and scope of your analysis, you may want to use only a single variant per model. This can be challenging to locate as not all variants have all the experiments and models. However, now that we have removed the incomplete results, we can now call the `remove_ensembles()` function which will only keep the *smallest* `member_id` for each model group. By smallest, we mean that first entry after a hierarchical sort using the integer index values of each label in the `member_id`.

```{code-cell}
cat.remove_ensembles()
print(cat.model_groups())
```

Now the results are much more manageable and ready to be downloaded for use in your analysis.

## Feedback

What do you [think](https://github.com/esgf2-us/intake-esgf/issues/new?assignees=&labels=&projects=&template=feature_request.md&title=) of this interface? We have found that it saves our students days of work, but are interested in critical feedback. Can you think of simpler interface? Are there other analysis tasks that are painful and time consuming that we could automate?

0 comments on commit 6b674e8

Please sign in to comment.