Your privacy, your choice

We use essential cookies to make sure the site can function. We also use optional cookies for advertising, personalisation of content, usage analysis, and social media.

By accepting optional cookies, you consent to the processing of your personal data - including transfers to third parties. Some third parties are outside of the European Economic Area, with varying standards of data protection.

See our privacy policy for more information on the use of your personal data.

for further information and to change your choices.
Skip to main content

Table 2 This table shows the hyperparameter combinations we tested (in all permutations). The chosen values are in underlined bold. These were determined by best classification performance

From: Uncertainty-aware automatic TNM staging classification for [18F] Fluorodeoxyglucose PET-CT reports for lung cancer utilising transformer-based language models and multi-task learning

Hyperparameter

Values tested (best performing in underlined bold)

Learning rate

9e-6

1e-5

2e-5

3e-5

4e-5

5e-5

Epochs

2

3

4

5

6

7

Batch size

1

2

4

8

-

-

Dropout probability

0

0.1

0.2

0.3

0.4

0.5

Back to article page

BMC Medical Informatics and Decision Making

ISSN: 1472-6947

Contact us