Home | History | Annotate | Download | only in distributions 
      1 # Copyright 2016 The TensorFlow Authors. All Rights Reserved.
      2 #
      3 # Licensed under the Apache License, Version 2.0 (the "License");
      4 # you may not use this file except in compliance with the License.
      5 # You may obtain a copy of the License at
      6 #
      7 #     http://www.apache.org/licenses/LICENSE-2.0
      8 #
      9 # Unless required by applicable law or agreed to in writing, software
     10 # distributed under the License is distributed on an "AS IS" BASIS,
     11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     12 # See the License for the specific language governing permissions and
     13 # limitations under the License.
     14 # ==============================================================================
     15 """Classes representing statistical distributions and ops for working with them.
     16 
     17 See the @{$python/contrib.distributions} guide.
     18 """
     19 from __future__ import absolute_import
     20 from __future__ import division
     21 from __future__ import print_function
     22 
     23 # pylint: disable=unused-import,wildcard-import,line-too-long,g-importing-member
     24 
     25 from tensorflow.contrib.distributions.python.ops import bijectors
     26 from tensorflow.contrib.distributions.python.ops.autoregressive import *
     27 from tensorflow.contrib.distributions.python.ops.binomial import *
     28 from tensorflow.contrib.distributions.python.ops.cauchy import *
     29 from tensorflow.contrib.distributions.python.ops.chi2 import *
     30 from tensorflow.contrib.distributions.python.ops.conditional_distribution import *
     31 from tensorflow.contrib.distributions.python.ops.conditional_transformed_distribution import *
     32 from tensorflow.contrib.distributions.python.ops.deterministic import *
     33 from tensorflow.contrib.distributions.python.ops.distribution_util import fill_triangular
     34 from tensorflow.contrib.distributions.python.ops.distribution_util import matrix_diag_transform
     35 from tensorflow.contrib.distributions.python.ops.distribution_util import reduce_weighted_logsumexp
     36 from tensorflow.contrib.distributions.python.ops.distribution_util import softplus_inverse
     37 from tensorflow.contrib.distributions.python.ops.distribution_util import tridiag
     38 from tensorflow.contrib.distributions.python.ops.estimator import *
     39 from tensorflow.contrib.distributions.python.ops.geometric import *
     40 from tensorflow.contrib.distributions.python.ops.half_normal import *
     41 from tensorflow.contrib.distributions.python.ops.independent import *
     42 from tensorflow.contrib.distributions.python.ops.inverse_gamma import *
     43 from tensorflow.contrib.distributions.python.ops.kumaraswamy import *
     44 from tensorflow.contrib.distributions.python.ops.logistic import *
     45 from tensorflow.contrib.distributions.python.ops.mixture import *
     46 from tensorflow.contrib.distributions.python.ops.mixture_same_family import *
     47 from tensorflow.contrib.distributions.python.ops.moving_stats import *
     48 from tensorflow.contrib.distributions.python.ops.mvn_diag import *
     49 from tensorflow.contrib.distributions.python.ops.mvn_diag_plus_low_rank import *
     50 from tensorflow.contrib.distributions.python.ops.mvn_full_covariance import *
     51 from tensorflow.contrib.distributions.python.ops.mvn_tril import *
     52 from tensorflow.contrib.distributions.python.ops.negative_binomial import *
     53 from tensorflow.contrib.distributions.python.ops.normal_conjugate_posteriors import *
     54 from tensorflow.contrib.distributions.python.ops.onehot_categorical import *
     55 from tensorflow.contrib.distributions.python.ops.poisson import *
     56 from tensorflow.contrib.distributions.python.ops.poisson_lognormal import *
     57 from tensorflow.contrib.distributions.python.ops.quantized_distribution import *
     58 from tensorflow.contrib.distributions.python.ops.relaxed_bernoulli import *
     59 from tensorflow.contrib.distributions.python.ops.relaxed_onehot_categorical import *
     60 from tensorflow.contrib.distributions.python.ops.sample_stats import *
     61 from tensorflow.contrib.distributions.python.ops.sinh_arcsinh import *
     62 from tensorflow.contrib.distributions.python.ops.test_util import *
     63 from tensorflow.contrib.distributions.python.ops.vector_diffeomixture import *
     64 from tensorflow.contrib.distributions.python.ops.vector_exponential_diag import *
     65 from tensorflow.contrib.distributions.python.ops.vector_laplace_diag import *
     66 from tensorflow.contrib.distributions.python.ops.vector_sinh_arcsinh_diag import *
     67 from tensorflow.contrib.distributions.python.ops.wishart import *
     68 from tensorflow.python.ops.distributions.bernoulli import *
     69 from tensorflow.python.ops.distributions.beta import *
     70 from tensorflow.python.ops.distributions.categorical import *
     71 from tensorflow.python.ops.distributions.dirichlet import *
     72 from tensorflow.python.ops.distributions.dirichlet_multinomial import *
     73 from tensorflow.python.ops.distributions.distribution import *
     74 from tensorflow.python.ops.distributions.exponential import *
     75 from tensorflow.python.ops.distributions.gamma import *
     76 from tensorflow.python.ops.distributions.kullback_leibler import *
     77 from tensorflow.python.ops.distributions.laplace import *
     78 from tensorflow.python.ops.distributions.multinomial import *
     79 from tensorflow.python.ops.distributions.normal import *
     80 from tensorflow.python.ops.distributions.student_t import *
     81 from tensorflow.python.ops.distributions.transformed_distribution import *
     82 from tensorflow.python.ops.distributions.uniform import *
     83 
     84 # pylint: enable=unused-import,wildcard-import,line-too-long,g-importing-member
     85 
     86 from tensorflow.python.util.all_util import remove_undocumented
     87 
     88 _allowed_symbols = [
     89     'auto_correlation',
     90     'bijectors',
     91     'Cauchy',
     92     'ConditionalDistribution',
     93     'ConditionalTransformedDistribution',
     94     'FULLY_REPARAMETERIZED',
     95     'NOT_REPARAMETERIZED',
     96     'ReparameterizationType',
     97     'Distribution',
     98     'Autoregressive',
     99     'Binomial',
    100     'Bernoulli',
    101     'Beta',
    102     'BetaWithSoftplusConcentration',
    103     'Categorical',
    104     'Chi2',
    105     'Chi2WithAbsDf',
    106     'Deterministic',
    107     'VectorDeterministic',
    108     'Exponential',
    109     'ExponentialWithSoftplusRate',
    110     'VectorExponentialDiag',
    111     'Gamma',
    112     'GammaWithSoftplusConcentrationRate',
    113     'Geometric',
    114     'HalfNormal',
    115     'Independent',
    116     'InverseGamma',
    117     'InverseGammaWithSoftplusConcentrationRate',
    118     'Kumaraswamy',
    119     'Laplace',
    120     'LaplaceWithSoftplusScale',
    121     'Logistic',
    122     'NegativeBinomial',
    123     'Normal',
    124     'NormalWithSoftplusScale',
    125     'Poisson',
    126     'PoissonLogNormalQuadratureCompound',
    127     'SinhArcsinh',
    128     'StudentT',
    129     'StudentTWithAbsDfSoftplusScale',
    130     'Uniform',
    131     'MultivariateNormalDiag',
    132     'MultivariateNormalFullCovariance',
    133     'MultivariateNormalTriL',
    134     'MultivariateNormalDiagPlusLowRank',
    135     'MultivariateNormalDiagWithSoftplusScale',
    136     'Dirichlet',
    137     'DirichletMultinomial',
    138     'Multinomial',
    139     'VectorDiffeomixture',
    140     'VectorLaplaceDiag',
    141     'VectorSinhArcsinhDiag',
    142     'WishartCholesky',
    143     'WishartFull',
    144     'TransformedDistribution',
    145     'QuantizedDistribution',
    146     'Mixture',
    147     'MixtureSameFamily',
    148     'ExpRelaxedOneHotCategorical',
    149     'OneHotCategorical',
    150     'RelaxedBernoulli',
    151     'RelaxedOneHotCategorical',
    152     'kl_divergence',
    153     'RegisterKL',
    154     'fill_triangular',
    155     'matrix_diag_transform',
    156     'reduce_weighted_logsumexp',
    157     'softplus_inverse',
    158     'tridiag',
    159     'normal_conjugates_known_scale_posterior',
    160     'normal_conjugates_known_scale_predictive',
    161     'percentile',
    162     'assign_moving_mean_variance',
    163     'assign_log_moving_mean_exp',
    164     'moving_mean_variance',
    165     'estimator_head_distribution_regression',
    166     'quadrature_scheme_softmaxnormal_gauss_hermite',
    167     'quadrature_scheme_softmaxnormal_quantiles',
    168     'quadrature_scheme_lognormal_gauss_hermite',
    169     'quadrature_scheme_lognormal_quantiles',
    170 ]
    171 
    172 remove_undocumented(__name__, _allowed_symbols)
    173