Everything runs, but the precision recall f1 calculation is horrendously slow. Need to find a smarter way to do that.