Skip to content

Commit

Permalink
Readme update
Browse files Browse the repository at this point in the history
  • Loading branch information
kahst committed Oct 19, 2023
1 parent 52039a7 commit 67ee757
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.adoc
Original file line number Diff line number Diff line change
Expand Up @@ -686,9 +686,10 @@ Here is a list of all command line arguments:
--mixup, Whether to use mixup for training.
--upsampling_ratio, Balance train data and upsample minority classes. Values between 0 and 1. Defaults to 0.
--upsampling_mode, Upsampling mode. Can be 'repeat', 'mean' or 'smote'. Defaults to 'repeat'.
--model_format, Model output format. Can be 'tflite', 'raven' or 'both'. Defaults to 'tflite'.
----
+
. After training, you can use the custom trained classifier with the `--classifier` argument of the `analyze.py` script.
. After training, you can use the custom trained classifier with the `--classifier` argument of the `analyze.py` script. If you want to use the custom classifier in Raven, make sure to set `--model_format raven`.
+
NOTE: Adjusting hyperparameters (e.g., number of hidden units, learning rate, etc.) can have a big impact on the performance of the classifier.
We recommend trying different hyperparameter settings.
Expand Down

0 comments on commit 67ee757

Please sign in to comment.