| { | |
| "model_name": "LSTM_CDRs", | |
| "framework": "TensorFlow/Keras", | |
| "task": "cdr-amino-acid-sequence-generation", | |
| "description": "LSTM_CDRs is an LSTM/GRU-based CDR amino acid sequence generation model for training on CDR sequence clusters and sampling new candidate sequences.", | |
| "entry_points": { | |
| "train": "LSTM_CDRs.py", | |
| "inference": "LSTM_CDRs.py", | |
| "smoke_test": "python LSTM_CDRs.py --help" | |
| }, | |
| "data_dir": "data", | |
| "datasets": [ | |
| "data/Cluster1.csv", | |
| "data/Cluster2.csv", | |
| "data/Cluster3.csv", | |
| "data/Cluster4.csv" | |
| ], | |
| "requirements": "requirements.txt", | |
| "environment": "environment.yml", | |
| "weight_dir": "<run_name>/checkpoint", | |
| "license": "MIT" | |
| } | |