Skip to content

Add Set2Set Aggregation#6

Open
MarcS18 wants to merge 3 commits intochemprop:mainfrom
MarcS18:main
Open

Add Set2Set Aggregation#6
MarcS18 wants to merge 3 commits intochemprop:mainfrom
MarcS18:main

Conversation

@MarcS18
Copy link
Copy Markdown

@MarcS18 MarcS18 commented Mar 19, 2026

Description

This PR adds a Chemprop compatible version of the Set2Set aggregation function [1]. Set2Set is an aggregation function designed to map a set of input vectors to a single, permutation-invariant graph-level or set-level embedding. It is used in molecular property prediction tasks as a readout function to aggregate node-level information, ensuring the output is invariant to the input order of atoms.

This code is tested and working with Chemprop 2.2.1

Questions

None

Checklist

  • License file included
  • Documentation provided (perhaps a Notebook or README)
  • Tests provided and passing on the stated Chemprop version

@JacksonBurns JacksonBurns self-requested a review March 26, 2026 14:16
@JacksonBurns
Copy link
Copy Markdown
Member

Hi thanks for the contrbution @MarcS18 , I will get to reviewing this!

fix equation formatting on github
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