Skip to content

Conversation

@aditya0by0
Copy link
Member

@aditya0by0 aditya0by0 commented Dec 6, 2025

Use new prediction pipeline as per ChEB-AI/python-chebai#135

@aditya0by0
Copy link
Member Author

@aditya0by0
Copy link
Member Author

This is implement in ChEB-AI/python-chebai#135.

  • The target labels for each trained model can also be included in lightning checkpoint. This avoid to manual bookkeeping of the labels file for each model-data variant. If this is agreed upon, we can create a separate issue for this.

TODO:

remove target_labels_path: classes.txt from registry
update readme
check if logic needs any change for classes enabled in checkpoing

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