K -- The length of a chain.

Running the file train.py can produce the results for mTRE in the setting K=5.

Running the file train1.py can produce the results for mTRE in the setting K=10.