We pass the column for the labels, the features, and the grouping
column as arguments to the function now.

This was done for some experiments with using only the software
features for classification.