Updated code to output tuned sbir dataset