Views
No views yet
FlyBaseGeneAbstractClassifier, a machine learning model designed to classify gene-paper abstract pairs into two labels for Drosophila genes:transformers library and was trained on a system with the following hardware:1from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
2
3# Initialize the tokenizer and the model
4tokenizer = AutoTokenizer.from_pretrained("scibert", model_max_length=512)
5model = AutoModelForSequenceClassification.from_pretrained("cgrivaz/FlyBaseGeneAbstractClassifier", num_labels=2)
6
7# Create a pipeline
8model_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)