search

WayScience / phenotypic_profiling

Machine learning for predicting 15 single-cell phenotypes from cell morphology profiles
Creative Commons Attribution 4.0 International
2 stars 3 forks source link

Train module refactor #20

Closed roshankern closed 1 year ago

roshankern commented 1 year ago

This PR is ready for review!

In this PR, a model is created for each combination of:

model_types = ["final", "shuffled_baseline"]
feature_types = ["CP", "DP", "CP_and_DP"]

these combinations are tested in a for loop to better reuse code. Each model is saved to the models/ directory.

Also, the 1.split_data module has a one-line change to include Greg's suggestion from https://github.com/WayScience/phenotypic_profiling_model/commit/2f50b408e3448251863eb5497b50cceeb66159c2#r103899488