nepp1d0 commited on
Commit
42068d1
·
1 Parent(s): 2428fa2

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
README.md ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Prot_bert finetuned on GPCR_train dataset of Drug Target prediction
2
+
3
+ Trainig paramenters:
4
+ overwrite_output_dir=True,
5
+ evaluation_strategy="epoch",
6
+ learning_rate=1e-3,
7
+ weight_decay=0.001,
8
+ per_device_train_batch_size=batch_size,
9
+ per_device_eval_batch_size=batch_size,
10
+ push_to_hub=True,
11
+ fp16=True,
12
+ logging_steps=logging_steps,
13
+ save_strategy='epoch',
14
+ num_train_epochs=2