rdolphin commited on
Commit
2ce57c1
1 Parent(s): 9fc9c4e

Update README.md

Browse files

multi_class has been deprecated in favor of multi_label. Updating example code accordingly.

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -43,11 +43,11 @@ classifier(sequence_to_classify, candidate_labels)
43
  # 'sequence': 'one day I will see the world'}
44
  ```
45
 
46
- If more than one candidate label can be correct, pass `multi_class=True` to calculate each class independently:
47
 
48
  ```python
49
  candidate_labels = ['travel', 'cooking', 'dancing', 'exploration']
50
- classifier(sequence_to_classify, candidate_labels, multi_class=True)
51
  #{'labels': ['travel', 'exploration', 'dancing', 'cooking'],
52
  # 'scores': [0.9945111274719238,
53
  # 0.9383890628814697,
43
  # 'sequence': 'one day I will see the world'}
44
  ```
45
 
46
+ If more than one candidate label can be correct, pass `multi_label=True` to calculate each class independently:
47
 
48
  ```python
49
  candidate_labels = ['travel', 'cooking', 'dancing', 'exploration']
50
+ classifier(sequence_to_classify, candidate_labels, multi_label=True)
51
  #{'labels': ['travel', 'exploration', 'dancing', 'cooking'],
52
  # 'scores': [0.9945111274719238,
53
  # 0.9383890628814697,