code: def predict(self, mllib_data):
          
          if isinstance(mllib_data, pyspark.mllib.linalg.Matrix):
              return to_matrix(self._master_network.predict(from_matrix(mllib_data)))
          elif isinstance(mllib_data, pyspark.mllib.linalg.Vector):
              return to_vector(self._master_network.predict(from_vector(mllib_data)))
          else:
              raise ValueError(
                  'Provide either an MLLib matrix or vector, got {}'.format(mllib_data.__name__))
docs: Predict probabilities for an RDD of features