Python extract_txt_arr示例

示例#1

文件： input_data.py 项目： GritBear/tensorflow-experiment

def read_input():
    trainFile = "train.csv"
    testFile = "test.csv"

    print("Reading Training Sets")
    TrainData, header = io_util.extract_txt_arr(trainFile)
    Train_set = TrainData[:, 1:-1]
    Train_target = TrainData[:, -1]

    print("Reading Submission Sets")
    TestData, header = io_util.extract_txt_arr(testFile)
    Test_set = TestData[:, 1:]

    # process training data
    print("Compose DataSets")
    dataSet = DataSet(Train_set, Train_target, Test_set)

    return dataSet

示例#2

文件： test_script.py 项目： GritBear/tensorflow-experiment

import io_util

trainFile = 'train.csv'
testFile = 'test.csv'

TrainInput, header = io_util.extract_txt_arr(trainFile)

print header
print TrainInput

ID = TrainInput[:,0]
Target = TrainInput[:,-1]
Train = TrainInput[:,1:-1]

print ID
print Target
print Train

TestInput, header = io_util.extract_txt_arr(testFile)