Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Update experiments README
Browse files Browse the repository at this point in the history
ivan-chai committed Jan 13, 2025
1 parent 2366099 commit c511524
Showing 2 changed files with 6 additions and 1 deletion.
5 changes: 5 additions & 0 deletions experiments/transactions-agepred/README.md
Original file line number Diff line number Diff line change
@@ -23,3 +23,8 @@ Set the number of threads:
```
spark-submit --master 'local[8]'
```

# Predict embeddings
```bash
python3 -m hotpp.embed --config-dir configs --config-name detection +embeddings_path=embeddings.parquet
```
2 changes: 1 addition & 1 deletion experiments/transactions-mbd/README.md
Original file line number Diff line number Diff line change
@@ -10,5 +10,5 @@ python3 -m hotpp.train --config-dir configs --config-name detection

# Predict embeddings
```bash
python3 -m hotpp.embed --config-dir configs --config-name detection ~logger +embeddings_path=embeddings-last-1000.pkl trainer.devices=1 ~data_module.train_params'
python3 -m hotpp.embed --config-dir configs --config-name detection +embeddings_path=embeddings.parquet trainer.devices=1
```

0 comments on commit c511524

Please sign in to comment.