Added a way to collect transition probability training data.