Exemplo n.º 1
0
def getInputData(prefixes, table, sqlFilter):
    fieldList = genMagByPrefix(prefixes)
    fieldList.append("`type`")
    fieldList.append("mainclass")
    sql_select = sqlHelper.genSelectSQLWithFieldList(fieldList, table, sqlFilter)

    outFile = "data_input.txt"
    outFile_tag = "data_input_tag.txt"
    output = open(outFile, "w")
    output_tag = open(outFile_tag, "w")
    count = 0

    res = []
    tags = []
    data = sqlHelper.getFetchAll(sql_select)
    for item in data:
        record = []
        colCount = len(item)
        for i in range(colCount - 1):
            record.append(item[i])

        res.append(record)
        tags.append(item[colCount - 1])

        output.write(util.listToString(record))
        output.write("\n")
        output_tag.write(item[colCount - 1])
        output_tag.write("\n")
        count += 1

    print "input data count : %s" % count
    output.close()
    output_tag.close()
    return np.mat(res), np.mat(tags).T