利用rnn进行irris分类

import numpy as np
from random import shuffle
import tensorflow as tf

train_path = 'C:/Users/user/Documents/irris_train.txt'
test_path = 'C:/Users/user/Documents/irris_test.txt'
logs_path = 'C:/Users/user/Desktop/log'

def get_files(file_dir):
    irris_zero = []
    label_irris_zero = []
    irris_one = []
    label_irris_one = []
    
    file = open(file_dir)
    for line in file:
        line = line.replace('\n','')
        wordList = line.split('\t')
        temp = []
        for word in wordList[:-1]:
            temp.append(float(word))
        if int(wordList[-1]) == 0:
            irris_zero.append(temp)
            label_irris_zero.append(int(0))
        else:
            irris_one.append(temp)
            label_irris_one.append(int(1))
    
    image_list = np.array(irris_zero+irris_one)
    label_list = np.array(label_irris_zero