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()
