Add class prediction methods to InferenceModel

[KevinMusgrave]

Basically something like sklearn KNeighborsClassifier. We could also add nearest-centroid classification and others.

See #342 and #343