configuration.json

{
  "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"
}