如何在张量流中训练线性回归模型时确定分类特征中值的相应权重？

Question

我已经使用 tensorflow 训练了一个线性回归模型。我有数字和分类特征。对于这个例子，其中一个分类特征是“has_Alt”，它的值为 0 或 1。当我尝试获取与此特征对应的权重值时，我得到一个二维向量——[[0.23342037 ] [0.22178642]]。

我如何知道所提供功能中的值 0 和 1 对应的权重。

可能会说第一个是 0，第二个是 1。但是，如果值是绿色和蓝色会怎样。那时我们怎么知道？下面是我如何添加功能的代码 sn-p。

def get_feature_columns(dftrain):
    CATEGORICAL_COLUMNS = ['hasAlt']

    feature_columns = []
    for feature_name in CATEGORICAL_COLUMNS:
        vocabulary = dftrain[feature_name].unique()
        feature_columns.append(tf.feature_column.categorical_column_with_vocabulary_list(feature_name, vocabulary))
    return feature_columns

Answer 1

如果你能分享完整的代码，答案会更清楚。

但是，已经创建了示例代码以更好地解释。这些值取决于它在vocabulary 中出现的顺序。

例如：

代码： 如果分类列是绿色和蓝色。

dftrain = {'hasAlt': ['green','blue','blue','blue','blue','blue','green','green']}
CATEGORICAL_COLUMNS = ['hasAlt']

feature_columns = []
for feature_name in CATEGORICAL_COLUMNS:
    vocabulary = set(dftrain[feature_name])
    feature_columns.append(tf.feature_column.categorical_column_with_vocabulary_list(feature_name, vocabulary))

print("Vocabulary order is :",vocabulary)
feature_columns

输出 -

Vocabulary order is : {'blue', 'green'}
[VocabularyListCategoricalColumn(key='hasAlt', vocabulary_list=('blue', 'green'), dtype=tf.string, default_value=-1, num_oov_buckets=0)]

代码： 如果分类列是 0 和 1。

dftrain = {'hasAlt': [1,0,0,0,0,1,1,1]}
CATEGORICAL_COLUMNS = ['hasAlt']

feature_columns = []
for feature_name in CATEGORICAL_COLUMNS:
    vocabulary = set(dftrain[feature_name])
    feature_columns.append(tf.feature_column.categorical_column_with_vocabulary_list(feature_name, vocabulary))

print("Vocabulary order is :",vocabulary)
feature_columns

输出 -

Vocabulary order is : {0, 1}
[VocabularyListCategoricalColumn(key='hasAlt', vocabulary_list=(0, 1), dtype=tf.string, default_value=-1, num_oov_buckets=0)]

如果还不清楚，请分享完整代码以便更好地解释。

希望这能回答您的问题。快乐学习。