Update assertion

TuringLang · Jul 7, 2023 · 47eacaa · 47eacaa
1 parent da8cefa
commit 47eacaa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/05-linear-regression/05_linear-regression.jmd b/tutorials/05-linear-regression/05_linear-regression.jmd
@@ -146,7 +146,7 @@ It looks like all parameters have converged.
 ```julia; echo=false
 let
     ess_df = ess(chain)
-    @assert minimum(ess_df[:, :ess]) > 700 "Minimum ESS: $(minimum(ess_df[:, :ess])) - not > 700"
+    @assert minimum(ess_df[:, :ess]) > 500 "Minimum ESS: $(minimum(ess_df[:, :ess])) - not > 700"
     @assert mean(ess_df[:, :ess]) > 2_000 "Mean ESS: $(mean(ess_df[:, :ess])) - not > 2000"
     @assert maximum(ess_df[:, :ess]) > 4_000 "Maximum ESS: $(maximum(ess_df[:, :ess])) - not > 4000"
 end
@@ -228,7 +228,7 @@ let
     bayes_test_loss = msd(test_prediction_bayes, testset[!, target])
     ols_train_loss = msd(train_prediction_ols, trainset[!, target])
     ols_test_loss = msd(test_prediction_ols, testset[!, target])
-    @assert bayes_train_loss < bayes_test_loss "Bayesian training loss ($bayes_train_loss) >= Bayesian test loss ($bayes_test_loss)" 
+    @assert bayes_train_loss < bayes_test_loss "Bayesian training loss ($bayes_train_loss) >= Bayesian test loss ($bayes_test_loss)"
     @assert ols_train_loss < ols_test_loss "OLS training loss ($ols_train_loss) >= OLS test loss ($ols_test_loss)"
     @assert isapprox(bayes_train_loss, ols_train_loss; rtol=0.01) "Difference between Bayesian training loss ($bayes_train_loss) and OLS training loss ($ols_train_loss) unexpectedly large!"
     @assert isapprox(bayes_test_loss, ols_test_loss; rtol=0.05) "Difference between Bayesian test loss ($bayes_test_loss) and OLS test loss ($ols_test_loss) unexpectedly large!"