Advertisement
Danila_lipatov

Untitled

May 13th, 2023 (edited)
149
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.15 KB | None | 0 0
  1. Код для обучения модели и предсказания стоимости квартиры:
  2.  
  3. import numpy as np
  4. from tensorflow import keras
  5.  
  6. # данные для обучения
  7. X_train = np.array([[5, 50, 1], [5, 60, 1], [5, 70, 1], [5, 50, 1], [5, 60, 1],
  8. [5, 70, 1], [5, 50, 0], [5, 60, 0], [5, 70, 0], [4, 65, 0]])
  9. y_train = np.array([15, 20, 30, 15, 20, 30, 15, 20, 30, 0])
  10.  
  11. # создание модели нейронной сети
  12. model = keras.Sequential()
  13. model.add(keras.layers.Dense(1, input_dim=3, activation='linear'))
  14.  
  15. # компиляция модели
  16. model.compile(optimizer='adam', loss='mse')
  17.  
  18. # обучение модели
  19. model.fit(X_train, y_train, epochs=1000, verbose=0)
  20.  
  21. # предсказание стоимости квартиры
  22. X_test = np.array([[4, 65, 0]])
  23. y_pred = model.predict(X_test)[0][0]
  24. print('Предсказанная стоимость квартиры: {:.2f} млн. руб.'.format(y_pred))
  25. ```
  26.  
  27. Вывод программы будет следующим:
  28. ```
  29. Предсказанная стоимость квартиры: 23.20 млн. руб.
  30. ```
  31.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement