定义一个线性回归模型:linear_regression_model.py
import tensorflow as tf
import numpy as np
class linearRegressionModel:
def __init__(self,x_dimen):
self.x_dimen = x_dimen
self._index_in_epoch = 0
self.constructModel()
self.sess = tf.Session()
self.sess.run(tf.global_variables_initializer())
#权重初始化
def weight_variable(self,shape):
initial = tf.truncated_normal(shape,stddev = 0.1)
return tf.Variable(initial)
#偏置项初始化
def bias_variable(self,shape):
initial = tf.constant(0.1,shape=shape)
return tf.Variable(initial)
#每次选取100个样本,如果选完,重新打乱
def next_batch(self,batch_size):
start = self._index_in_epoch
self._index_in_epoch += batch_size
if self._index_in_epoch > self._num_datas: #如果选完
perm = np.arange(self._num_datas)
np.random.shuffle(perm) #打乱数据