epam / epam/Indigo

GMPP: Add convenient main function

Open
#629 0 comments 0 reactions 1 assignee Claimed by @suppnova View on GitHub
ML Priority: High python
Dominant language
C++
Stars
406
Forks
134
Avg merge
2d 11h
Merged PRs (30d)
24

Description

Blocked by #602.

Add GMPP main function that:
1. Takes GNN model type (like MPNN or GAT) and optionally some kwargs parameters for it.
2. Takes featurisers preset (for atoms and bonds).
3. Takes CSV dataset, SMILES column and Assay column.
4. Builds model, performs cross-validation and returns train, validation and test score.
5. Use it in the notebook.

Going extra mile:
* Add CLI interface using `argparse`.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.