|
|
@@ -5,7 +5,7 @@ import sys
|
|
|
class DSSMReader(MultiSlotDataGenerator):
|
|
|
def __init__(self):
|
|
|
super(DSSMReader, self).__init__()
|
|
|
- self.feature_dim = 5 # 设置特征维度
|
|
|
+ self.feature_dim = 157 # 设置特征维度
|
|
|
self.data_count = 0 # 添加计数器
|
|
|
def init(self, config=None):
|
|
|
pass
|
|
|
@@ -24,7 +24,7 @@ class DSSMReader(MultiSlotDataGenerator):
|
|
|
left_features = [float(x) for x in left_features.split(',')]
|
|
|
right_features = [float(x) for x in right_features.split(',')]
|
|
|
|
|
|
- sys.stderr.write(f"left_features length: {len(left_features)}\n")
|
|
|
+
|
|
|
# 构建输出列表
|
|
|
output = []
|
|
|
#output.append(("sample_id", [sample_id])) # 样本ID
|
often