Tree Features during testing

[maxfscher]

Hi,

first of all thank you very much for this great repository. Could you kindly provide any insights on how tree features can be used in the "attention_map.py" file during testing?

Best Max