Handle accuracy of small input datasets better