【北京大学】人工智能实践：Tensorflow笔记 4-5

generateds.py         生成数据

import numpy as np
import matplotlib.pyplot as plt
seed = 2
def generateds():
    rdm = np.random.RandomState(seed)
    X  = rdm.randn(300,2)
    Y_ = [int(x0*x0+x1*x1<2)for (x0,x1) in X]
    # 1红0 绿
    Y_c = [['red' if y else 'blue']for y in Y_]
    # 形状整理，-1表示跟随第二列计算，第二个表示多少列
    X = np.vstack(X).reshape(-1,2)
    Y_ = np.vstack(Y_).reshape(-1, 1)
    return X,Y_,Y_c
# def draw(X,Y_c):
#     x =[None]*300
#     y =[None]*300
#     i = 0
#     for x1,y1 in X:
#         x[i] = x1
#         y[i] =y1
#         i+=1
#     print(Y_c)
#     z = [None]*300
#     for i in range(300):
#         z[i]="".join(Y_c[i])
#     print(z)
#     plt.figure()
#     plt.scatter(x,y,c=z)
#     plt.show()

# X,Y_,Y_c = generateds()
# draw(X,Y_c)
 

forward.py  搭建网络，设计网络结构

import tensorflow as tf
def forward(x,regularizer)