excluding 0 prob transitions: -6.5172343254089355
-4 ll penalty: -6.405089855194092
-5 ll penalty: -6.44964075088501
-6 ll penalty: -6.4941911697387695