search

Freshwind-Bioinformatics / TABR-BERT

TABR-BERT: an Accurate and Robust BERT-based Transfer Learning Model for TCR-pMHC Interaction Prediction
Other
8 stars 1 forks source link

Why does an input file of the predict model require a label columns? #3

Closed RysBen closed 3 months ago

RysBen commented 3 months ago

Dear developer,

The below was some content from README.md:

Usage: predict_tcr_pmhc_binding.py [options]
Required:
      --input STRING: The data to be predicted (*.csv) 
                      Required columns: ["allele", "peptide", "cdr3"]

When I used an input files with three columns ("allele", "peptide", "cdr3"), some error happened:

Traceback (most recent call last):
  File "/.../predict_tcr_pmhc_binding.py", line 199, in <module>
    labels = input_df["label"].tolist()
  File "/.../envs/tabr_bert/lib/python3.9/site-packages/pandas/core/frame.py", line 3024, in __getitem__
    indexer = self.columns.get_loc(key)
  File "/.../envs/tabr_bert/lib/python3.9/site-packages/pandas/core/indexes/base.py", line 3082, in get_loc
    raise KeyError(key) from err
KeyError: 'label'

So why does an input file of the predict model require a label column?

Thanks, Ben

RysBen commented 3 months ago

Sorry, I forgot I cancel annotation of the label line ⊙︿⊙