Commit 860d8495 authored by 郭羽's avatar 郭羽

排序模型优化

parent bfaec84e
...@@ -144,7 +144,7 @@ def predict(model_path,df): ...@@ -144,7 +144,7 @@ def predict(model_path,df):
# model = tf.keras.models.model_from_json(model.to_json) # model = tf.keras.models.model_from_json(model.to_json)
n = 1000 n = 1000
dd = dict(df.sample(n=n)) dd = dict(df.sample(n=n))
for i in range(10): for i in range(5):
timestmp1 = int(round(time.time() * 1000)) timestmp1 = int(round(time.time() * 1000))
model_new.predict(dd, batch_size=10000) model_new.predict(dd, batch_size=10000)
print("测试样本数:{},测试耗时ms:{}".format(n, int(round(time.time() * 1000)) - timestmp1)) print("测试样本数:{},测试耗时ms:{}".format(n, int(round(time.time() * 1000)) - timestmp1))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment