예제 #1
0
파일: main.py 프로젝트: Nuomisu/cs5242
def q2_4(data, path):
    case = int(sys.argv[1])
    wfile = "w-100-40-4.csv"
    bfile = "b-100-40-4.csv"
    if case == 2:
        wfile = "w-28-6-4.csv"
        bfile = "b-28-6-4.csv"
    elif case == 3:
        wfile = "w-14-28-4.csv"
        bfile = "b-14-28-4.csv"
    w, b = loadwb(path + wfile, path + bfile)
    model = three.ThreeLayers(case, 14, 4, w, b)
    data["x_train"] = data["x_a"]
    data["y_train"] = data["y_a"]
    exp = Experiment(model, data, 1, 1)
    w, b = exp.check_speicalpoint()
    # output dw db
    with open("result/" + wfile, 'wb') as csvfile:
        writer = csv.writer(csvfile, delimiter=',')
        for wouter in w:
            for line in wouter:
                writer.writerow(line)
    with open("result/" + bfile, 'wb') as csvfile:
        writer = csv.writer(csvfile, delimiter=',')
        for line in b:
            writer.writerow(line)


if __name__ == "__main__":
    x_train, y_train, x_test, y_test, x_a, y_a = readfile("data")
    data = {