Skip to content

Commit 3659143

Browse files
doc: CLarify random sampling instructions
Extended the instructions to make it clear that the user must always explicitly include the background class.
1 parent 602271d commit 3659143

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

weight_sampling_tutorial.ipynb

+3-1
Original file line numberDiff line numberDiff line change
@@ -297,7 +297,9 @@
297297
"n_classes_source = 81\n",
298298
"# TODO: Set the indices of the classes that you want to pick for the sub-sampled weight tensors.\n",
299299
"# In case you would like to just randomly sample a certain number of classes, you can just set\n",
300-
"# `classes_of_interest` to an integer instead of the list below.\n",
300+
"# `classes_of_interest` to an integer instead of the list below. Either way, don't forget to\n",
301+
"# include the background class. That is, if you set an integer, and you want `n` positive classes,\n",
302+
"# then you must set `classes_of_interest = n + 1`.\n",
301303
"classes_of_interest = [0, 3, 8, 1, 2, 10, 4, 6, 12]\n",
302304
"# classes_of_interest = 9 # Uncomment this in case you want to just randomly sub-sample the last axis instead of providing a list of indices.\n",
303305
"\n",

0 commit comments

Comments
 (0)