1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning for Text with PyTorch
Connected

cvičení

Vyhodnocení výkonu modelu

Tým PyBooks udělal velký pokrok v doporučovacím systému pro knihy. Modelářský tým ti připravil dva různé modely pro tento systém. Jeden model je postavený na LSTM (lstm_model) a druhý používá GRU (gru_model). Tvým úkolem je tyto modely vyhodnotit a porovnat.

K dispozici máš testovací štítky y_test a predikce y_pred_lstm pro lstm_model a y_pred_gru pro gru_model.

Pokyny

100 XP
  • Definuj přesnost (accuracy), preciznost (precision), návratnost (recall) a F1 skóre pro vícetřídní klasifikaci zadáním parametrů num_classes a task.
  • Vypočítej a vypiš accuracy, precision, recall a F1 skóre pro lstm_model.
  • Stejným způsobem vypočítej hodnoticí metriky pro gru_model.