Skip to content

Model ensemble pull request#55

Merged
vxfung merged 19 commits intoFung-Lab:mainfrom
orivera2280:main
Jan 7, 2024
Merged

Model ensemble pull request#55
vxfung merged 19 commits intoFung-Lab:mainfrom
orivera2280:main

Conversation

@orivera2280
Copy link
Copy Markdown
Contributor

Most changes are in the base_trainer and property_trainer files. Generally, model ensemble methods are indicated by if statements checking for the type of model (if it is a list, it is a model ensemble) or by checking if the "model_ensemble" parameter from the config is > 1. I also made changes to the task.py file to support predictions on ensembles. These are indicated by the checks for the type of the data loader (if list, then it assumes a model ensemble is being used).

@vxfung vxfung merged commit 08894ab into Fung-Lab:main Jan 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants