[mlpack-git] [mlpack/mlpack] DatasetMapper & Imputer (#694)

[Keon Kim]

> +BOOST_AUTO_TEST_CASE(MeanImputationTest)
> +{
> +  arma::mat input("3.0 0.0 2.0 0.0;"
> +                  "5.0 6.0 0.0 6.0;"
> +                  "9.0 8.0 4.0 8.0;");
> +  arma::mat outputT; // assume input is transposed
> +  arma::mat output;  // assume input is not transposed
> +  double mappedValue = 0.0;
> +
> +  MeanImputation<double> imputer;
> +
> +  // transposed
> +  imputer.Apply(input, outputT, mappedValue, 0/*dimension*/, true);
> +
> +  BOOST_REQUIRE_CLOSE(outputT(0, 0), 3.0, 1e-5);
> +  BOOST_REQUIRE_CLOSE(outputT(0, 1), 2.5, 1e-5);

the current implementation applies only to the given dimension (0 in this case)
so, 0 is ignored and mean is going to be (3 + 2) / 2 = 2.5
and it replaces 0 with 2.5.

---
You are receiving this because you are subscribed to this thread.
Reply to this email directly or view it on GitHub:
https://github.com/mlpack/mlpack/pull/694/files/da4e40981c40e171858f41de4e85e08af2c7e48d..d8618ec909005b0453f0a3803ac1e6cc5e588f1f#r68699218
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mailman.cc.gatech.edu/pipermail/mlpack-git/attachments/20160627/d250a223/attachment.html>