gorkaartola's picture
Update tests.py
c9a76ea
test_cases = [
{
"predictions": [0.01, 0.01, 0.9],
"references": [2, 0],
},
{
"predictions": [0.01, 0.01, 0.9],
"references": [1, 2],
},
{
"predictions": [0.9, 0.01, 0.01],
"references": [0, 0],
}
{
"predictions": [0.9, 0.01, 0.01],
"references": [2, 2],
}
{
"predictions": [0.01, 0.9, 0.01],
"references": [1, 0],
}
{
"predictions": [0.01, 0.9, 0.01],
"references": [0, 2],
}
]