healthcare-chatbot

This model is a fine-tuned version of google/flan-t5-small on an unknown dataset. It achieves the following results on the evaluation set:

• Train Loss: 0.8977
• Validation Loss: 0.8199
• Epoch: 14

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• optimizer: {'name': 'AdamW', 'weight_decay': 0.01, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': {'module': None, 'class_name': 'WarmupSchedule', 'config': {'warmup_steps': 2472, 'post_warmup_schedule': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 22248, 'end_learning_rate': 1e-07, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}}, 'registered_name': 'WarmupSchedule'}, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
• training_precision: float32

Training results

Train Loss	Validation Loss	Epoch
8.0286	1.3975	0
1.3825	1.0703	1
1.1485	0.9700	2
1.0618	0.9292	3
1.0162	0.8996	4
0.9837	0.8785	5
0.9609	0.8628	6
0.9439	0.8505	7
0.9307	0.8414	8
0.9207	0.8347	9
0.9127	0.8290	10
0.9065	0.8248	11
0.9025	0.8222	12
0.8995	0.8204	13
0.8977	0.8199	14

Framework versions

• Transformers 4.57.0
• TensorFlow 2.19.0
• Datasets 4.0.0
• Tokenizers 0.22.1