diff --git a/README.md b/README.md index e288e36..aeaf8e8 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,9 @@ Output label is the probability of a paranormal subreddit. # Pytorch logistic regression The code can be found in Logistic.py + Trained models end with .pth extension. + Geval results: ``` @@ -22,6 +24,8 @@ Precision 0.6257 Recall 0.4094 ``` +Logs from training have been copy-pasted into `l1_epochs.txt` (for single-layer model) and `l2_epochs.txt (for two-layer model). + Sources -------