1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Úvod do hlubokého učení s Keras
Connected

Cvičení

Rozpoznávání číslic

Teď si postavíš model na datasetu digits – ukázkovém datasetu, který je předinstalovaný ve scikit-learn. Dataset digits obsahuje ručně psané číslice od 0 do 9 ve formátu 8×8 pixelů:

Cílem je rozpoznat každou z 10 možných číslic na základě obrázku – jde tedy o klasifikaci do více tříd.

Dataset je už rozdělený na X_train, y_train, X_test a y_test, přičemž 30 % dat tvoří testovací sadu. Popisky jsou už zakódované jako one-hot vektory, takže funkci to_categorical() z Kerasu nepotřebuješ.

Pojďme sestavit tento nový model!

Pokyny

100 XP
  • Přidej vrstvu Dense s 16 neurony, aktivací relu a parametrem input_shape nastaveným na celkový počet pixelů obrázku číslice 8×8.
  • Přidej vrstvu Dense s 10 výstupy a aktivací softmax.
  • Zkompiluj model s optimalizátorem adam, ztrátovou funkcí categorical_crossentropy a metrikou accuracy.
  • Ověř, že model funguje, a spusť predikci na X_train.