분자구조 이미지 SMILES 변환 AI 경진대회

베이스라인 코랩에서 돌려볼 수 있는 코드

2020.09.20 19:01 7,807 조회 language

안녕하세요.
코드를 돌리면서 사실 파인튜닝 하는 것보다 일단 베이스라인 따라가는 것도 참 힘들다는 생각이 들었습니다.
기존에 자원이 없는 분들에게 조금이나마 도움이 될까 싶어서 코랩과 같은 클라우드 환경에서 돌릴 수 있는 코드를 공유합니다.
기존 인셉션 피쳐맵을 저장하는 방식이 아니라 인코더에서 바로 만드는 방법이라 하드 자원이 많이 필요 없어도 되고.
트레이닝 셋 용량도 크기 때문에 데이콘 코드를 참고하여 smiles문자열에서 이미지를 만드는 방식으로 바꾸었습니다.
학습 데이터셋의 100분의 1만 사용했기 때문에 실제로 사용하실 때는 이거보다 많은 데이터를 사용하시면 더 좋을 듯 합니다.

코드
로그인이 필요합니다
0 / 1000
당쇠
2020.09.20 23:32

오옷 감사하므니다.!

다담
2020.09.21 09:59

감사합니다 !! 

포도당
2020.09.21 11:28

저만 베이스라인도 어렵다는 생각한게 아니군요!! 감사합니다!!

김웅곤
2020.09.22 17:06

최고입니다!!!