2016-08-05 6 views
3

저는 인공 지능과 tensorflow가있는 전체 시작 자입니다. 따라서 이것이 바보 같은 질문 인 경우 용서해주십시오. 나는이 자습서를 기반으로하는 스크립트를 사용하여 tensorflow 네트워크를 훈련했습니다Tensorflow 'features'형식

https://www.tensorflow.org/versions/r0.10/tutorials/wide_and_deep/index.html

내가 훈련 괜찮다고 생각합니다. 는 지금은 하나의 입력에 대한 예측을하기 위해이 방법을 실행하는 whant :

tf.contrib.learn.DNNClassifier.predict_proba(x=x) 

하지만 난 "X"매개 변수를 구축하는 방법에 대한 문서를 찾을 수 없습니다 ... 나는 tryed :

x = {k: tf.SparseTensor(indices=[[0, 0]], values=[d_data[k]], shape=[1, 1]) for k in COLUMNS} 

여기서, d_data은 약 150 개의 키/값 쌍을 포함하는 사전입니다. 은 필요한 모든 키가있는 목록입니다. 동일한 설정이 네트워크를 학습하는 데 사용되었습니다.

그러나 오류 가지고 :

AttributeError: 'dict' object has no attribute 'dtype' 

그래서 ... x는 '딕셔너리'안을 ...하지만 그 다음해야 하는가? 누구든지 내게 어떤 지시를 내릴 수 있습니까?

고마워요.

답변

2

BaseEstimator 클래스는 documentation이 좋습니다.

x: Matrix of shape [n_samples, n_features...]. Can be iterator that returns arrays of features. The training input samples for fitting the model. If set, `input_fn` must be `None`. 

여기서 문서를 수정하려고합니다. 지적 해 주셔서 고마워요.

0

나는 동일한 오류가 발생했다. 그러나 이것은 우리가 이전 버전의 tensorflow (나는 0.8.0에있다)를 사용하고 있기 때문에 적합 방법이 이제는 다른 입력 유형 인 input_fn을 취할 수 있다고 생각한다. 사전의 형태는, 그 X 같은 텐서 행렬 객체 입력 필수 이유 때문에,이 기능은 'input_fn을'이없는 내 현재 릴리스 here

def fit(self, x=None, y=None, input_fn=None, steps=None, batch_size=None, 
     monitors=None, max_steps=None): 

참조.

그 동안 해결책을 찾을 수 있었습니까?