9d5b280
1
2
3
4
5
6
7
8
import numpy as np def doc_to_target(doc): return np.round(((doc["toxicity_ai"] + doc["toxicity_human"]) > 5.5), 0).astype( np.int32 )
