classifier_tf.py 文件源码

def prediction_time(self, X):
        start_time = time.time()
        dp = {}
        for key, value in X.items():
            dp[key] = value[np.newaxis, 0]
        times = 100
        for _ in range(times):
            self.predict_proba(dp)
        return (time.time() - start_time) / times