remove one_hot, first try

I added a y_labels return value here because if it is not provided then we don't know the ranging scale of y, which is provided in the initial implementation. But if we add the new y_labels parameter in the __init__ method, I think that also works.