Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/QuantEcon/Expectations.jl

Expectation operators for Distributions.jl objects
https://github.com/QuantEcon/Expectations.jl

distributions julia quadrature statistics

Last synced: about 2 months ago
JSON representation

Expectation operators for Distributions.jl objects

Awesome Lists containing this project

README

        

[![CI](https://github.com/QuantEcon/Expectations.jl/workflows/CI/badge.svg)](https://github.com/QuantEcon/Expectations.jl/actions?query=workflow%3ACI)
[![codecov](https://codecov.io/gh/QuantEcon/Expectations.jl/branch/master/graph/badge.svg)](https://codecov.io/gh/QuantEcon/Expectations.jl)

[![](https://img.shields.io/badge/docs-latest-blue.svg)](https://QuantEcon.github.io/Expectations.jl/dev)

# Expectations

Installation (for Julia v1.0 and up):
```julia
pkg> add Expectations
```
See [Pkg docs for more details](https://julialang.github.io/Pkg.jl/v1/managing-packages/#Adding-packages-1)

This is a package designed to simplify the process of taking expectations of functions of random variables.

### Expectation Operator

The key object is the `expectation` function, which returns an operator:

```julia
dist = Normal()
E = expectation(dist)
E(x -> x)
```
For convenience, the operator can be applied directly to a function instead of being cached,
```julia
expectation(x->x^2, dist)
```

As a linear operator on vectors using the nodes of the distribution
```julia
dist = Normal()
E = expectation(dist)
x = nodes(E)
f(x) = x^2
E * f.(x) == dot(f.(x), weights(E))
```

### Random Variables

The underlying distributions are objects from `Distributions.jl` (currently `<:UnivariateDistribution`).

**Starting with 1.3.0, we also support mixture models.**

### Quadrature Algorithms

We support different types of Gaussian quadrature (Gauss-Hermite, Gauss-Legendre, Gauss-Laguerre, etc.) based on the distribution, as well as some methods with user-defined nodes (e.g., trapezoidal integration).

We have rules for the following distributions:

* Normal
* ChiSq
* LogNormal
* Exponential
* Beta
* Gamma/Erlang
* Uniform
* Continuous Univariate (compact; generic fallback)
* Continuous Univariate (no restriction; approximates with quantile grid)
* Discrete

See docs for more info.

### Mixture Models

We also support mixture models, e.g.

```julia
d = MixtureModel([Uniform(), Normal(), Gamma()]);
E = expectation(d);
E(x -> x) # 0.5000000000000016
```

The `MixtureExpectation` objects support most of the same behavior as the individual `IterableExpectation`s.

```julia
2E(x -> x) # 1.000000000000003
weights(E) # [1/3, 1/3, 1/3]
expectations(E) # component expectations
```