variational_approx_example using small dataset (for memory check)