def layer_sizes(X, Y):
    """
    设置网络结构

    Args:
        X: 输入的数据
        Y: 输出值

    Return:
        n_x: 输入层节点数
        n_h: 隐藏层节点数
        n_y: 输出层节点数
    """
    n_x = X.shape[0]  # 输入层大小（节点数）
    n_h = 4
    n_y = Y.shape[0]  # 输出层大小（节点数）
    return (n_x, n_h, n_y)