File: multilabel_test.py

package info (click to toggle)
python-pycm 4.5-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 4,044 kB
  • sloc: python: 5,361; sh: 8; makefile: 6
file content (35 lines) | stat: -rw-r--r-- 1,194 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# -*- coding: utf-8 -*-
"""
>>> from pycm import *
>>> mlcm = MultiLabelCM([{'dog'}, {'cat', 'dog'}], [{'cat'}, {'cat'}])
>>> len(mlcm)
2
>>> print(mlcm)
pycm.MultiLabelCM(classes: ['cat', 'dog'])
>>> mlcm
pycm.MultiLabelCM(classes: ['cat', 'dog'])
>>> mlcm.get_cm_by_sample(0)
pycm.ConfusionMatrix(classes: [0, 1])
>>> mlcm.get_cm_by_sample(0)
pycm.ConfusionMatrix(classes: [0, 1])
>>> mlcm.get_cm_by_sample(1)
pycm.ConfusionMatrix(classes: [0, 1])
>>> mlcm.get_cm_by_class('cat')
pycm.ConfusionMatrix(classes: [0, 1])
>>> mlcm.get_cm_by_class('cat')
pycm.ConfusionMatrix(classes: [0, 1])
>>> mlcm.get_cm_by_class('dog')
pycm.ConfusionMatrix(classes: [0, 1])
>>> mlcm.samplewise_cms[0] != mlcm.samplewise_cms[1]
True
>>> mlcm = MultiLabelCM([{'dog'}, {'cat', 'dog'}], [{'cat'}, {'cat', 'bird'}], classes=['dog', 'cat'])
>>> mlcm.get_cm_by_class('cat')
pycm.ConfusionMatrix(classes: [0, 1])
>>> mlcm.get_cm_by_class('dog')
pycm.ConfusionMatrix(classes: [0, 1])
>>> mlcm.classwise_cms['dog'] != mlcm.classwise_cms['cat']
True
>>> mlcm2 = MultiLabelCM([{'dog'}, {'cat', 'dog'}], [{'cat'}, {'cat'}], sample_weight=[2, 5])
>>> mlcm2.get_cm_by_class("dog")
pycm.ConfusionMatrix(classes: [0, 1])
"""