Добавьте взвешивание к выводу в кератах - PullRequest
Новая
       10

Добавьте взвешивание к выводу в кератах

0 голосов
/ 09 октября 2018

Я хочу добавить простое взвешивание к моему выводу в кератах.У меня есть эта модель: 

model = keras.Sequential([
    keras.layers.Dense(32, input_shape=(1,))
    # Hidden layers
    keras.layers.Dense(2))
])

На вход х случайное число с плавающей точкой от 1 до 0 Выход должен быть [0, 1], если х> 0,9 еще [1, 0]

Но если я начну обучать сеть 

x = np.random.random(1000)
y = (x > 0.9).astype(np.float32)
model.fit(x, y)

Она просто научится выводить это: 

   >>> model.predict(np.array([[1]]))<br>
   >>> [ [0.2] ]

Так что я думаю, что решение будет иметь какое-то взвешивание на выходе.Как мне это сделать?

...