Skip to content
Snippets Groups Projects
Commit b144db33 authored by Jörg Martin's avatar Jörg Martin
Browse files

Fixed evaluate_tabular

parent 52526a39
No related branches found
No related tags found
No related merge requests found
......@@ -144,7 +144,7 @@ def collect_metrics(x,y, seed=0,
noise_state = net.noise_is_on
net.train()
net.noise_on()
not_averaged_predictions = net.predict(x, number_of_draws=noneiv_number_of_draws,
not_averaged_predictions = net.predict(x, number_of_draws=eiv_number_of_draws,
take_average_of_prediction=False)
eiv_mean = torch.mean(not_averaged_predictions[0], dim=1)
if len(y.shape) <= 1:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment