English
medical
brain-data
mri
jesseab commited on
Commit
ae80538
·
1 Parent(s): f5c5cd5
README.md CHANGED
@@ -51,6 +51,12 @@ nohup python brain2vec.py train \
51
  --output_dir ./ae_output \
52
  --n_epochs 10 \
53
  > train_log.txt 2>&1 &
 
 
 
 
 
 
54
  ```
55
 
56
  # Methods
 
51
  --output_dir ./ae_output \
52
  --n_epochs 10 \
53
  > train_log.txt 2>&1 &
54
+
55
+ # run model inference to create *_embeddings.npz files
56
+ python brain2vec.py infererence \
57
+ --dataset_csv home/ubuntu/brain2vec/inputs.csv \
58
+ --aekl_ckpt /home/ubuntu/brain2vec/autoencoder_final.pth \
59
+ --output_dir /home/ubuntu/brain2vec
60
  ```
61
 
62
  # Methods
__pycache__/brain2vec.cpython-310.pyc ADDED
Binary file (18.8 kB). View file