|
@@ -20,6 +20,8 @@ sparse_features = [
|
|
|
"cid", "adid", "adverid",
|
|
|
"region", "city", "brand",
|
|
|
"vid", "cate1", "cate2",
|
|
|
+ "apptype", "hour", "hour_quarter", "root_source_scene", "root_source_channel", "is_first_layer", "title_split",
|
|
|
+ "profession"
|
|
|
]
|
|
|
tag_features = [
|
|
|
"user_vid_return_tags_2h", "user_vid_return_tags_1d", "user_vid_return_tags_3d",
|
|
@@ -127,6 +129,7 @@ for name in tag_features + seq_features:
|
|
|
separator: ','
|
|
|
}}""")
|
|
|
|
|
|
+
|
|
|
def wide_and_deep():
|
|
|
print("""
|
|
|
model_config {
|
|
@@ -161,6 +164,7 @@ model_config {
|
|
|
embedding_regularization: 1e-6
|
|
|
}""")
|
|
|
|
|
|
+
|
|
|
def deep_fm():
|
|
|
print("""
|
|
|
model_config {
|
|
@@ -196,6 +200,7 @@ model_config {
|
|
|
embedding_regularization: 1e-6
|
|
|
}""")
|
|
|
|
|
|
+
|
|
|
def fm():
|
|
|
print("""
|
|
|
model_config {
|
|
@@ -221,6 +226,7 @@ model_config {
|
|
|
embedding_regularization: 1e-5
|
|
|
}""")
|
|
|
|
|
|
+
|
|
|
def config_export():
|
|
|
print("""
|
|
|
export_config {
|