fixed minor bug in vigra_machine_learning.py