7 lines
324 B
Python
7 lines
324 B
Python
from collections import namedtuple
|
|
|
|
# 使用具名元组定义特征标记
|
|
SparseFeat = namedtuple('SparseFeat', ['name', 'vocabulary_size', 'embedding_dim'])
|
|
DenseFeat = namedtuple('DenseFeat', ['name', 'dimension'])
|
|
VarLenSparseFeat = namedtuple('VarLenSparseFeat', ['name', 'vocabulary_size', 'embedding_dim', 'maxlen'])
|