Skip to content

F
ffm-baseline
Commit d2540fea authored by Your Name's avatar Your Name
Browse files
Options

bug fix

parent e6876cb1
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
eda/esmm/Model_pipline/dist_predict.py
View file @ d2540fea
...@@ -5,6 +5,7 @@ import pytispark.pytispark as pti ...@@ -5,6 +5,7 @@ import pytispark.pytispark as pti
from pyspark.sql import SparkSession from pyspark.sql import SparkSession
import datetime import datetime
import pandas as pd import pandas as pd
from datetime import date, timedelta
import time import time
from pyspark import StorageLevel from pyspark import StorageLevel
import os import os
...@@ -142,8 +143,8 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False): ...@@ -142,8 +143,8 @@ def input_fn(filenames, batch_size=32, num_epochs=1, perform_shuffle=False):
return batch_features, batch_labels return batch_features, batch_labels
def main(): def main():
# dt_dir = (date.today() + timedelta(-1)).strftime('%Y%m%d') dt_dir = (date.today() + timedelta(-1)).strftime('%Y%m%d')
model_dir = "/home/gmuser/esmm/model_ckpt/DeepCvrMTL/" model_dir = "/home/gmuser/esmm/model_ckpt/DeepCvrMTL/" + dt_dir
te_files = ["hdfs://172.16.32.4:8020/strategy/esmm/nearby/part-r-00000"] te_files = ["hdfs://172.16.32.4:8020/strategy/esmm/nearby/part-r-00000"]
model_params = { model_params = {
"field_size": 15, "field_size": 15,
......
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