ALIGNN-BERT-TL codebase
This repository contains the code for Hybrid-LLM-GNN transfer learning framework to predict materials properties using concatenated embeddings extracted from ALIGNN and BERT models. The code provides the following functions:Generate string representations of provided crystal samples using NLP tools: Robocystallographer and ChemNLPUse a pre-trained BERT/MatBERT langugae model to extract context-aware word embeddings from text descriptions of a given crystal structureConcatenate LLM embeddings with ALIGNN embeddings obtained from feature-extraction-based ALIGNNTLPost-analysis script for model performance analysis and text-based model explanation<br>
Publication Details
- Journal
- Figshare
- Published
- 2024-10-06
- DOI
- https://doi.org/10.6084/m9.figshare.27162276
- Primary Topic
- Machine Learning in Materials Science
- Type
- article
- Field-Weighted Citation Impact
- 0.00