apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make suggested changes.

Browse Source
This commit is contained in:
akhandait
2018-08-06 23:29:36 +05:30
parent dcda11b786
commit a092dcc572
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/ann/dists/bernoulli_distribution_impl.hpp
+6 -4
View File
@@ -47,9 +47,11 @@ BernoulliDistribution<DataType>::BernoulliDistribution(
template<typename DataType>
DataType BernoulliDistribution<DataType>::Sample() const
{
DataType sample;
sample = arma::conv_to<DataType>::from(arma::randu<DataType>
(probability.n_rows, probability.n_cols) <= probability);
DataType sample = arma::randu<DataType>
(probability.n_rows, probability.n_cols);
for (size_t i = 0; i < sample.n_elem; i++)
sample(i) = sample(i) < probability(i);
return sample;
}
@@ -69,7 +71,7 @@ void BernoulliDistribution<DataType>::LogProbBackward(
if (!applyLogistic)
{
output = observation / (probability + eps) - (1 - observation) /
(1 - probability);
(1 - probability + eps);
}
else
{