Add an explanation on -l option
@@ -28,10 +28,25 @@ and outputs trained parameters and other information to a specified directory.
We specify an ID of GPU in use by `-g` or `--gpu` option.
Negative value indicate running the code with CPU.
The output directory can be specified by `-o` option.
-The default value is `result`.
+Its default value is `result`.
+The Tox21 dataset consists of several assays.
+Some molecules can have more than one types of assay results.
+We can specify which assay to use by specifying an assay name with `-l` option.
+Assay names are available by running the script with `-h` or `--help`
+or execute the following command:
+
+```
+python -c import chainerchem; chainerchem.datasets.get_tox21_label_names()
+```
+
+If `-l` option is not specified, this script conducts multitask learning with all labels.
The full options available including `-g` and `-o` are found
-by running the following command `python train_tox21.py --help`.
+by running the following command:
+
+```
+python train_tox21.py -h
+```
### Inference with a trained model
@@ -53,4 +68,4 @@ The prediction results are saved in the current directory as a `npy` file.
As with training, we can specify GPU/CPU to use by `-g` option.
The full options available including `-g` and `-i` are found
-by running the following command `python inference_tox21.py --help`.
+by running the following command `python inference_tox21.py -h`.
