machine learning working on a single batch, need to change to work on tracks not just segments