import tensorflow as tf
from random import shuffle#随机打乱,提高准确率
TRAINING_RATIO = 0.5
training_num = int(TRAINING_RATIO * len(outputs))
rand_idx = list(range(len(inputs))); shuffle(rand_idx)
inputs = inputs[rand_idx]; outputs = outputs[rand_idx]
x_train, y_train = inputs[:training_num], outputs[:training_num]
x_test, y_test = inputs[training_num:], outputs[training_num:]
class MyModel(tf.keras.Model):
def __init__(self, ):
super(MyModel, self).__init__()
self.w1 = tf.keras.layers.Dense(6, activation='tanh', use_bias=False)
self.w3 = tf.keras.layers.Dense(1, activation='linear', use_bias=False)
def call(self, inputs):
x = self.w1(inputs)
x = self.w3(x)
return tf.math.reduce_sum(x, axis=1,)