Skip to content

Commit ace84fa

Browse files
author
tixhjq
committed
fix example error
1 parent 3ed5b98 commit ace84fa

File tree

5 files changed

+10
-15
lines changed

5 files changed

+10
-15
lines changed

example/ctr_example/gsp_seq.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
pre_weight=None,sample_num=5)
4444
sparseDf,sparseInfo=data_pre.sparse_fea_deal(sparseDf)
4545

46-
train_df,test_df,y_train,y_test=data_pre.extract_train_test(
46+
train,val=data_pre.extract_train_test(
4747
targetDf=targetDf,test_idx=test_idx,train_idx=train_idx,sparseDf=sparseDf,seqDf=seqDf)
4848

4949
behaviorFea=['item_seq','gsp_seq']
@@ -52,4 +52,4 @@
5252
print(model.summary())
5353

5454
model.compile(loss="mean_squared_error",optimizer='adam',metrics=['accuracy'])
55-
model.fit(train_df,y_train,validation_data=(test_df,y_test),epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)])
55+
model.fit(train,validation_data=val,epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)])

example/ctr_example/seq.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
pre_weight=None,sample_num=5)
4242
sparseDf,sparseInfo=data_pre.sparse_fea_deal(sparseDf)
4343

44-
train,test=data_pre.extract_train_test(
44+
train,val=data_pre.extract_train_test(
4545
targetDf=targetDf,test_idx=test_idx,train_idx=train_idx,sparseDf=sparseDf,seqDf=seqDf)
4646

4747
candidateFea=['item_id','item_cate']
@@ -50,4 +50,4 @@
5050
model=DIEN(data_pre.FeatureInput(sparseInfo=sparseInfo,seqInfo=seqInfo),candidateFea=candidateFea,behaviorFea=behaviorFea)
5151
print(model.summary())
5252
model.compile(loss=tf.losses.binary_crossentropy,optimizer='adam',metrics=[tf.keras.metrics.AUC()])
53-
model.fit(train,validation_data=test,epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)])
53+
model.fit(train,validation_data=val,epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)])

example/ctr_example/session.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
sparseDf,sparseInfo=data_pre.sparse_fea_deal(sparseDf)
5050
denseDf,denseInfo=data_pre.dense_fea_deal(denseDf)
5151

52-
train_df,test_df,y_train,y_test=data_pre.extract_train_test(
52+
train,val=data_pre.extract_train_test(
5353
targetDf=targetDf,test_idx=test_idx,train_idx=train_idx,sparseDf=sparseDf,seqDf=seqDf,denseDf=denseDf)
5454

5555
candidateFea=['vid']
@@ -58,4 +58,4 @@
5858
model=DSIN(data_pre.FeatureInput(sparseInfo=sparseInfo,seqInfo=seqInfo),candidateFea=candidateFea,behaviorFea=behaviorFea)
5959
print(model.summary())
6060
model.compile(loss=tf.losses.binary_crossentropy,optimizer='adam',metrics=[tf.keras.metrics.AUC()])
61-
model.fit(train_df,y_train,validation_data=(test_df,y_test),epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)])
61+
model.fit(train,validation_data=val,epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)])

example/ctr_example/timeInterval.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
is_str=True,sample_num=5)
4242
sparseDf,sparseInfo=data_pre.sparse_fea_deal(sparseDf)
4343

44-
train_df,test_df,y_train,y_test=data_pre.extract_train_test(
44+
train,val=data_pre.extract_train_test(
4545
targetDf=targetDf,test_idx=test_idx,train_idx=train_idx,sparseDf=sparseDf,seqDf=seqDf)
4646

4747

@@ -54,7 +54,4 @@
5454
timestampFea=timestampFea,behaviorFea=behaviorFea,targetFea=targetFea)
5555
print(model.summary())
5656
model.compile(loss=tf.losses.binary_crossentropy,optimizer='adam',metrics=[tf.keras.metrics.AUC()])
57-
58-
from tensorflow.keras.callbacks import EarlyStopping
59-
model.fit(train_df,y_train,validation_data=(test_df,y_test),epochs=100,
60-
callbacks=[EarlyStopping(patience=10,verbose=5)])
57+
model.fit(train,validation_data=val,epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)])

example/ctr_example/un_seq.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,12 +53,10 @@
5353

5454
train_df,test_df,y_train,y_test=prepare_tool.extract_train_test(train_idx=train_idx,test_idx=test_idx,sparseDf=sparseDf,denseDf=denseDf,targetDf=targetDf,use_softmax=True)
5555
# train_df,test_df,y_train,y_test=prepare_tool.extract_train_test(train_idx=train_idx,test_idx=test_idx,sparseDf=sparseDf,targetDf=targetDf)
56-
train_x,train_y,val_set=prepare_tool.split_val_set(train_df,y_train,train_index,val_index)
56+
train,val=prepare_tool.split_val_set(train_df,y_train,train_index,val_index)
5757
#----------------------------train model--------------------------------------
5858

5959
model=FM(prepare_tool.FeatureInput(sparseInfo=sparseInfo,denseInfo=denseInfo,useAddLinear=False,useLinear=True,useFlattenLinear=False))
6060
print(model.summary())
6161
model.compile(loss=tf.losses.binary_crossentropy,optimizer='adam',metrics=[tf.keras.metrics.AUC()])
62-
model.fit(train_x,train_y,validation_data=val_set,batch_size=64,epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)],shuffle=False)
63-
model.predict(test_df)
64-
62+
model.fit(train,validation_data=val,epochs=100,callbacks=[tf.keras.callbacks.EarlyStopping(patience=10,verbose=5)])

0 commit comments

Comments
 (0)