EXA Avoiding TypeError when using --all_categories in plot_document_classification_20newsgroups.py (#12770)

dafeda · amueller · commit f772b98dab43 · 2018-12-14T11:43:46.000-05:00
diff --git a/examples/text/plot_document_classification_20newsgroups.py b/examples/text/plot_document_classification_20newsgroups.py
@@ -145,7 +145,7 @@ def size_mb(docs):
     len(data_train.data), data_train_size_mb))
 print("%d documents - %0.3fMB (test set)" % (
     len(data_test.data), data_test_size_mb))
-print("%d categories" % len(categories))
+print("%d categories" % len(target_names))
 print()
 
 # split a training set and a test set

Original file line number	Diff line number	Diff line change
`@@ -145,7 +145,7 @@ def size_mb(docs):`
`145`	`145`	`len(data_train.data), data_train_size_mb))`
`146`	`146`	`print("%d documents - %0.3fMB (test set)" % (`
`147`	`147`	`len(data_test.data), data_test_size_mb))`
`148`		`-print("%d categories" % len(categories))`
	`148`	`+print("%d categories" % len(target_names))`
`149`	`149`	`print()`
`150`	`150`
`151`	`151`	`# split a training set and a test set`