You are on page 1of 1

import tensorflow as tf

# Crear los datos de entrenamiento


x_train = [1.0, 2.0, 3.0, 4.0]
y_train = [0.0, -1.0, -2.0, -3.0]

# Crear el modelo de la red neuronal


model = tf.keras.models.Sequential([
tf.keras.layers.Dense(units=1, input_shape=[1])
])

# Compilar el modelo
model.compile(optimizer='sgd', loss='mean_squared_error')

# Entrenar el modelo
model.fit(x_train, y_train, epochs=100)

# Hacer una predicción


print(model.predict([5.0]))

///////// Evolución //////////////////////////

import tensorflow as tf

# Crear los datos de entrenamiento


x_train = [1.0, 2.0, 3.0, 4.0]
y_train = [0.0, -1.0, -2.0, -3.0]

# Crear el modelo de la red neuronal


model = tf.keras.models.Sequential([
tf.keras.layers.Dense(units=64, activation='relu', input_shape=[1]),
tf.keras.layers.Dense(units=1)
])

# Compilar el modelo
optimizer = tf.keras.optimizers.Adam(learning_rate=0.01)
model.compile(optimizer=optimizer, loss='mean_squared_error',
metrics=['mse', 'mae'])

# Entrenar el modelo
model.fit(x_train, y_train, epochs=1000, verbose=0)

# Hacer una predicción


print(model.predict([5.0]))

You might also like