7123df402c43e1537f73f63db5d90ed4ea03009d,cde/tests/unittests_estimators.py,TestRiskMeasures,test_mean_mixture,#TestRiskMeasures#,159

Before Change

    X = data[:, 0:2]
    Y = data[:, 2:4]

    model = MixtureDensityNetwork(n_centers=5)
    model.fit(X, Y)

    mean_est = model.mean_(x_cond=np.array([[1,2]]))
    self.assertAlmostEqual(mean_est[0][0], 7, places=0)

After Change

    X = data[:, 0:2]
    Y = data[:, 2:4]

    model = MixtureDensityNetwork("mdn_mean", 2, 2,n_centers=5)
    model.fit(X, Y)

    mean_est = model.mean_(x_cond=np.array([[1,2]]))
    self.assertAlmostEqual(mean_est[0][0], 7, places=0)

Instances