TF Learnを利用してDNNClassifierを実装しようと考えております。
Yターゲットの分類結果を1、2の2種類だとエラーにならず
1,2,3だとNanLossDuringTrainingErrorエラーになります。
分類数を増やすには、どのようにすれば良いかご教授いただければ幸いです。
python
1# -*- coding: utf-8 -*- 2 3from tensorflow.contrib import learn 4import numpy as np 5 6 7x_data = np.empty((0,4), float) 8x_data = np.append(x_data, np.array([[1, 1, 1, 1.1]]), axis=0) 9x_data = np.append(x_data, np.array([[1, 1, 1, 1.2]]), axis=0) 10x_data = np.append(x_data, np.array([[1, 1, 1, 1.3]]), axis=0) 11x_data = np.append(x_data, np.array([[1, 1, 1, 1.4]]), axis=0) 12 13#このYターゲットの分類結果を1、2の2種類だとエラーにならず 14#1,2,3だとエラーになる。 15#y_target = np.array([1,1,2,2]) # OK 16y_target = np.array([1,1,2,3]) # NanLossDuringTrainingError 17 18 19classifier = learn.DNNClassifier(hidden_units=[10,20,10], n_classes=3) 20classifier.fit(x=x_data, y=y_target, steps=20) 21 22print("end") 23 24# Error tensorflow.contrib.learn.python.learn.monitors.NanLossDuringTrainingError: NaN loss during training.
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。