nflows 다운로드 nflows 소스 코드 다운로드

nflows

파이썬

v0.14

다운로드

nflows

nflows Pytorch를 사용하여 정상화 흐름의 포괄적 인 컬렉션입니다.

설치

PYPI에서

PYPI에서 설치하려면 :

 python -m pip install nflows

Conda-Forge에서

프로젝트 디렉토리 실행에서 pixi 사용하여 프로젝트에 nflows 설치하고 추가하려면

 pixi add nflows

활성화 된 환경에서 conda 사용하여 특정 콘다 환경에 설치하려면

 conda install --channel conda-forge nflows

용법

흐름을 정의하려면 :

 from nflows import transforms , distributions , flows

# Define an invertible transformation.
transform = transforms . CompositeTransform ([
    transforms . MaskedAffineAutoregressiveTransform ( features = 2 , hidden_features = 4 ),
    transforms . RandomPermutation ( features = 2 )
])

# Define a base distribution.
base_distribution = distributions . StandardNormal ( shape = [ 2 ])


# Combine into a flow.
flow = flows . Flow ( transform = transform , distribution = base_distribution )

입력의 로그 확률을 평가하려면 :

 log_prob = flow . log_prob ( inputs )

흐름에서 샘플 :

 samples = flow . sample ( num_samples )

워크 플로의 추가 예는 예제 폴더에 제공됩니다.

개발

개발을위한 모든 종속성을 설치하려면 :

 pip install -r requirements.txt

Nflows를 인용합니다

패키지를 인용하려면 :

 @software { nflows ,
  author       = { Conor Durkan and
                  Artur Bekasov and
                  Iain Murray and
                  George Papamakarios } ,
  title        = { {nflows}: normalizing flows in {PyTorch} } ,
  month        = nov,
  year         = 2020 ,
  publisher    = { Zenodo } ,
  version      = { v0.14 } ,
  doi          = { 10.5281/zenodo.4296287 } ,
  url          = { https://doi.org/10.5281/zenodo.4296287 }
}