def predict()

in machine-learning/container/scripts/predictor.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

    def predict(cls,input):
        """For the input, do the predictions and return them.

        Args:
            input (a pandas dataframe): The data on which to do the
            predictions.

            There will be one prediction per row in the dataframe
        """

        sess = K.get_session()
        with sess.graph.as_default():
            clf = cls.get_model()
            return clf.predict(input)