import numpy as np
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
# Генерация обучающих данных: 1000 примеров с двумя входными числами
X = np.random.rand(1000, 2) # двумерный массив (1000, 2)
y = X.sum(axis=1) # сумма двух чисел по каждой строке, массив размером (1000,)
# Создание модели: один полносвязный слой, который возвращает одно число
model = Sequential([
Dense(1, input_shape=(2,))
])
# Компиляция модели с оптимизатором SGD (stochastic gradient descent) и функцией потерь MSE
model.compile(optimizer='sgd', loss='mse'
# Обучение модели (100 эпох, можно регулировать для улучшения точности)
model.fit(X, y, epochs=100, verbose=0)
# Тестирование модели: ожидаем, что сумма 0.3 и 0.7 ≈ 1.0
test_input = np.array([[0.3, 0.7]])
predicted_sum = model.predict(test_input)
print("Предсказанная сумма:", predicted_sum[0][0])