def predict_class(self, features):
if isinstance(features, RDD):<EOL><INDENT>return self.predict_class_distributed(features)<EOL><DEDENT>else:<EOL><INDENT>return self.predict_class_local(features)<EOL><DEDENT>
Model inference base on the given data which returning label
:param features: it can be a ndarray or list of ndarray for locally inference
                 or RDD[Sample] for running in distributed fashion
:return: ndarray or RDD[Sample] depend on the the type of features.