Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/EthanRosenthal/torchmf
matrix factorization in PyTorch
https://github.com/EthanRosenthal/torchmf
matrix-factorization pytorch recommender-systems
Last synced: 2 days ago
JSON representation
matrix factorization in PyTorch
- Host: GitHub
- URL: https://github.com/EthanRosenthal/torchmf
- Owner: EthanRosenthal
- Created: 2017-04-04T11:28:14.000Z (over 7 years ago)
- Default Branch: master
- Last Pushed: 2023-07-06T21:35:30.000Z (over 1 year ago)
- Last Synced: 2024-08-02T13:23:05.530Z (3 months ago)
- Topics: matrix-factorization, pytorch, recommender-systems
- Language: Python
- Homepage: https://www.ethanrosenthal.com/2017/06/20/matrix-factorization-in-pytorch/
- Size: 48.8 KB
- Stars: 127
- Watchers: 4
- Forks: 33
- Open Issues: 3
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# torchmf
matrix factorization in PyTorch
## Installation
### pip
```commandline
pip install -r requirements.txt
```
### conda
```commandline
conda env create -f environment.yml
conda activate torchmf
```
## Example
```bash
$ python run.py --example explicit
( 1 ): 100%|████████████████████████████████████████████████████████| 89/89 [00:01<00:00, 85.88it/s, train_loss=7.9]
Epoch: 1 train: 15.04790 val: 8.84972
( 2 ): 100%|███████████████████████████████████████████████████████| 89/89 [00:01<00:00, 84.06it/s, train_loss=2.96]
Epoch: 2 train: 4.34132 val: 4.04638
( 3 ): 100%|███████████████████████████████████████████████████████| 89/89 [00:01<00:00, 81.54it/s, train_loss=1.51]
Epoch: 3 train: 1.87918 val: 2.43315
( 4 ): 100%|███████████████████████████████████████████████████████| 89/89 [00:01<00:00, 85.55it/s, train_loss=1.19]
Epoch: 4 train: 1.21419 val: 1.80296
( 5 ): 100%|██████████████████████████████████████████████████████| 89/89 [00:00<00:00, 90.87it/s, train_loss=0.945]
Epoch: 5 train: 0.99693 val: 1.49770
( 6 ): 100%|██████████████████████████████████████████████████████| 89/89 [00:00<00:00, 89.33it/s, train_loss=0.914]
Epoch: 6 train: 0.90174 val: 1.33501
( 7 ): 100%|██████████████████████████████████████████████████████| 89/89 [00:00<00:00, 115.70it/s, train_loss=0.83]
Epoch: 7 train: 0.85230 val: 1.23783
( 8 ): 100%|██████████████████████████████████████████████████████| 89/89 [00:01<00:00, 88.85it/s, train_loss=0.879]
Epoch: 8 train: 0.82072 val: 1.17781
( 9 ): 100%|█████████████████████████████████████████████████████| 89/89 [00:00<00:00, 119.93it/s, train_loss=0.766]
Epoch: 9 train: 0.79898 val: 1.13976
(10 ): 100%|█████████████████████████████████████████████████████| 89/89 [00:00<00:00, 122.18it/s, train_loss=0.736]
Epoch: 10 train: 0.77820 val: 1.10951
```
which looks something like
