Function main

your_first_network.py:248–281

()

        
#-------------MAIN-------------------------
def main():
    global current_loss
    data = cp.creat_data(800) #生成数据
    cp.plot_data(data, "Right classification")

    #选择起始网络
    use_this_network = 'n' #No
    while use_this_network != 'Y' and use_this_network != 'y':
        network = Network(NETWORK_SHAPE)
        inputs = data[:, (0, 1)]
        outputs = network.network_forward(inputs)
        classification = classify(outputs[-1])
        data[:, 2] = classification
        cp.plot_data(data, "Choose network")
        use_this_network = input("Use this network? Y to yes, N to No \n")
    
    #进行训练
    do_train = input("Train? Y to yes, N to No \n")
    while do_train == 'Y' or do_train == 'y' or do_train.isnumeric() == True:
        if do_train.isnumeric() == True:
            n_entries = int(do_train)
        else:
            n_entries = int(input("Enter the number of data entries used to train. \n"))
            
        network.train(n_entries)
        do_train = input("Train? Y to yes, N to No \n")
        
    #演示训练效果
    inputs = data[:, (0, 1)]
    outputs = network.network_forward(inputs)
    classification = classify(outputs[-1])
    data[:, 2] = classification
    cp.plot_data(data, "After training")
    print("谢谢，再见！")
#----------------TEST-------------------------
def test():
    pass