modelの抽象化 - Githubissues

tubo213 / kaggle-child-mind-institute-detect-sleep-states

MIT License

115 stars 85 forks source link

modelの抽象化 #25

Closed tubo213 closed 11 months ago

tubo213 commented 11 months ago

AbstractModelをつくる

メソッド

forward: lossとlogitsの辞書を返す
predict: step毎の各イベントの生起確率を返す. 勾配の計算はしない

tubo213 commented 11 months ago

https://github.com/tubo213/kaggle-child-mind-institute-detect-sleep-states/pull/34