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