import tensorflow as tf from dataload import load_data from classify import Classifier # 데이터 준비 train_data, test_data = load_data( 'C:/Users/student/machineLearning/Tensorflow/iris.csv', 4, 0.6) # 노드 수 결정 num_input = train_data.x_data.shape[1] # 입력 노드 수 num_output = train_data.y_data.shape[1] # 출력 노드 수 num_hidden = 10 num_hidden2 = 20 # 분류기 생성 iris = Classifier(num_input, num_output, num_hidden, num_hidden2, 0.01) # 훈련 평가 iris.train(train_data, 1000, 100) iris.test(test_data) # 질의 answer = iris.query([[5.8, 4, 1.2, 0.2]]) print(answer) iris.close()