Usando Keras / Redes Neurais para dados categóricos

votos
2

No exemplo CNN para o conjunto de dados minst para Keras dizem-lhe como fazer uma boa rede CNN reconhecer mão dígitos escritos. A questão é que ele não lhe diz como prever novos dígitos.

Por exemplo dar uma imagem, se eu fizer isso:

model.predict(image)

em vez de me dizer o que dígitos que pensa que é, em vez disso me dá uma lista de 10 números (presumivelmente probabilidades)

Publicado 16/03/2016 em 12:56
fonte usuário
Em outras línguas...                            


1 respostas

votos
2

Você pode usar argmax de numpy para descobrir classe que tem probabilidade máxima

import numpy as np
probabilities = model.predict(image)
classes = np.argmax(probabilities, axis=-1)
Respondeu 16/03/2016 em 18:15
fonte usuário

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more