R/learnpython - My code is not working. Python

[PalPalash]

Hello guys ,
(BTW before using an IDE i was using Jupyter)

I am making a python project and this did not work out . Please help

Files : GitHub

(Code is not fully finished)

When i run:
python myTraining.py
it spits out this error:

Traceback (most recent call last):
  File "e:/WhiteHat JR/Hackathon/code/hackathon_whjr-covid-calculator-1/myTraining.py", line 25, in <module>
    Y_train = train[['infectionProb' ]].to_numpy().reshape(800,)
  File "C:\ProgramData\Anaconda3\lib\site-packages\pandas\core\frame.py", line 2908, in __getitem__
    indexer = self.loc._get_listlike_indexer(key, axis=1, raise_missing=True)[1]
  File "C:\ProgramData\Anaconda3\lib\site-packages\pandas\core\indexing.py", line 1254, in _get_listlike_indexer
    self._validate_read_indexer(keyarr, indexer, axis, raise_missing=raise_missing)
  File "C:\ProgramData\Anaconda3\lib\site-packages\pandas\core\indexing.py", line 1298, in _validate_read_indexer
    raise KeyError(f"None of [{key}] are in the [{axis_name}]")
KeyError: "None of [Index(['infectionProb'], dtype='object')] are in the [columns]"