Samsung AI Challenge for Scientific Discovery

Dacon 대회 참여 코드

Graph data로 변환

분자 데이터 정보가 smiles type으로 주어져있다. rdkit 라이브러리를 이용하여 분자 특성을 추출하여 pytorch geometric 의 graph type으로 변환해주었다.

Model

사용한 model 은 CGConv이다.
CGCN-리뷰
CGConv- PyG

run code

python main.py

학습결과

plot