Skip to content

S
serviceRec
Commit 66f2ea84 authored by 郭羽's avatar 郭羽
Browse files
Options

update feature

parent a70efdce
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
train/train_service.py
View file @ 66f2ea84
...@@ -92,9 +92,9 @@ def getWeight(x): ...@@ -92,9 +92,9 @@ def getWeight(x):
def getDataSet(df,shuffleSize = 10000,batchSize=128): def getDataSet(df,shuffleSize = 10000,batchSize=128):
# print(df.dtypes) # print(df.dtypes)
labels = df.pop('label') labels = df.pop('label')
df["page_stay"] = labels["label"].map(lambda x:1) # df["page_stay"] = df["page_stay"].map(getWeight)
weights = df.pop('page_stay') # weights = df.pop('page_stay')
dataSet = tf.data.Dataset.from_tensor_slices((dict(df), labels,weights)).shuffle(shuffleSize).batch(batchSize) dataSet = tf.data.Dataset.from_tensor_slices((dict(df), labels)).shuffle(shuffleSize).batch(batchSize)
return dataSet return dataSet
def getTrainColumns(train_columns,data_vocab): def getTrainColumns(train_columns,data_vocab):
......
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