-
Notifications
You must be signed in to change notification settings - Fork 1
/
Makefile
40 lines (36 loc) · 868 Bytes
/
Makefile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
train_agnews:
python train.py \
--use-dvat \
--n-labeled ${N_LABELED} \
--tsa 'lin_schedule' \
--confidence 0.0 \
--seed ${SEED} \
--normalize-grad 'inf' \
--learning-rate ${LR} \
--swap-ratio 0.25 \
--topk 10 \
--data-path data/AG_NEWS/
train_yahoo:
python train.py \
--use-dvat \
--n-labeled ${N_LABELED} \
--tsa 'lin_schedule' \
--confidence 0.0 \
--seed ${SEED} \
--normalize-grad 'inf' \
--learning-rate ${LR} \
--swap-ratio 0.25 \
--topk 10 \
--data-path data/YAHOO/
train_dbpedia:
python train.py \
--use-dvat \
--n-labeled ${N_LABELED} \
--tsa 'lin_schedule' \
--confidence 0.0 \
--seed ${SEED} \
--normalize-grad 'inf' \
--learning-rate ${LR} \
--swap-ratio 0.25 \
--topk 10 \
--data-path data/DBpedia/