DNNCLassifier Tensorflow上的标签键类型错误

Question

我想将标签嵌入Tensorflow中的DNNClassifier模型。与文档示例不同，在这里，我得到以下错误消息：

label_keys_values = ["satan", "ipsweep", "nmap", "portsweep"]  
m = tf.contrib.learn.DNNClassifier(model_dir=model_dir,
                                  feature_columns=deep_columns,
                                  n_classes=4,
                                  hidden_units=[12, 4],
                                  label_keys=label_keys_values)
m.fit(input_fn=train_input_fn, steps=200)

文件"embedding_model_probe.py"，第118行，m.fit（input_fn=train_input_fn，步骤=200）文件"/usr/本地/lib/python2.7/dist-包/Tenorflow/python/util/deprecation.py"，第281行，new_func返回func（*args，**kwargs）文件"/usr/local/lib/python2.7/dist-包/tenstorflow/contrib/学习/python/学习/估计器/estimator.py"，第430行，在适合损失=自我。_train_model（input_fn=input_fn，钩子=钩子）File"/usr/local/lib/python2.7/dist-包/tenstorflow/contrib/学习/python/学习/估计器/<--plhd--4_get_train_ops（功能，标签）File"/usr/local/lib/python2.7/dist-包/tenstorflow/contrib/学/python/学/估计器/estimator.py"，第1132行，_get_train_ops返回自己。_call_model_fn（功能，标签，model_fn_lib。ModeKeys.TRAIN）File"/usr/local/lib/python2.7/dist-包/tenstorflow/contrib/学/python/学/估计器/estimator.py"，第1103行，_call_model_fnmodel_fn_results=自。_model_fn（功能，标签，**kwargs）File"/usr/local/lib/python2.7/dist-包/tenstorflow/contrib/学/python/学/估计器/dnn.py"，第180行，_dnn_model_fnlogits=logits）File"/usr/local/lib/python2.7/dist-包/tenstorflow/contrib/学/python/学/估计器/head.py"，第1004行，create_model_fn_ops标签=自我。_transform_labels（模式=模式，标签=标签）File"/usr/本地/lib/python2.7/dist-包/tensorflow/contrib/学习/python/学习/估计器/head.py"，第1033行，_transform_labels"label_ids"：table.lookup（labels_tensor），File"/usr/loc/lib/python2.7/dist-包/tensorflow/contrib/lookup/lookup_ops.py"，第179行，在查找（自我。_key_dtype，keys.dtype））打字错误：签名不匹配。密钥必须是dtype

另一方面，如果我将label_key_values列设为numpy.array，则会得到以下错误：

label_keys_values = np.array(["satan", "ipsweep", "nmap", "portsweep"], dtype='string')

回溯（最后一次调用）：文件“Embedded_model_probe.py”，第116行，在label_keys=label_keys_values）文件/usr/local/lib/python2.7/dist packages/tensorflow/contrib/learn/python/estimators/dnn.py中，第337行，在init label_keys=label_keys）文件中“/usr/local/lib/python2。7/dist包/tensorflow/contrib/learn/python/learn/estimators/head。py“，第331行，在多类头标签（keys=标签（keys））文件“/usr/local/lib/python2中。7/dist包/tensorflow/contrib/learn/python/learn/estimators/head。py“，第986行，在init if label_keys和len（label_keys）！=n_classes:ValueError：具有多个元素的数组的真值不明确。请使用a.any（）或a.all（）

Answer 1

我找到了解决办法。正如官方文件所说：

"如果用户在构造函数中指定label_keys，则标签必须是来自label_keys词汇表的字符串。"

在我的例子中，我将训练和测试集中的标签列转换为一个热向量（整数值），并且label\u keys\u values数组中的值与它们不匹配。