Training Parameters


Generative AI Lab supports the Transfer Learning feature offered by Spark NLP for Healthcare 3.1.2. This feature is available for project manages and project owners, but only if a valid Healthcare NLP license is loaded into the Generative AI Lab.

In this case, the feature can be activated for any project by navigating to the Train page. It requires the presence of a base model trained with MedicalNERModel.

If a MedicalNER model is available on the Models Hub section of the Generative AI Lab, it can be chosen as a starting point of the training process. This means the base model will be Fine Tuned with the new training data.

When Fine Tuning is enabled, the same embeddings used for training the base model will be used to train the new model. Those need to be available on the Models Hub section as well. If present, embeddings will be automatically selected, otherwise users must go to the Models Hub page and download or upload them.

Last updated