Commit 004f276d authored by Carlos GO's avatar Carlos GO
Browse files

learning readme

parent fdd6f849
# Model training
A default run can be launched with the command below:
```
python learning/main.py
```
Running this command assumes that the `-da` option points to a folder which contains annotated graphs (networkx RNA graphs associated with a fingerprint vector).
See `data_processor.py` for details on creating annotated graphs.
Most important model parameters which can be specified from the command line:
* node embedding dimensions
* graph pooling function
* whether to warm start the model with a pre-trained embedder
For a full list of command line options run:
```
python learning/meain.py -h
```
All necessary files are saved in a folder inside `results/` with the run id specified at command ine (`-n`)
Once training is complete you can load the model to make new predictions.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment