esto no funciona
This commit is contained in:
parent
ddb9dc9576
commit
79f475e502
16
src/main.py
16
src/main.py
|
@ -72,6 +72,7 @@ def instantiate_model(A, index, pad_index, device):
|
||||||
print(cls)
|
print(cls)
|
||||||
return cls, phi
|
return cls, phi
|
||||||
|
|
||||||
|
|
||||||
def main(opt):
|
def main(opt):
|
||||||
|
|
||||||
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
|
||||||
|
@ -236,18 +237,3 @@ if __name__ == '__main__':
|
||||||
create_path_if_not_exists(opt.pickle)
|
create_path_if_not_exists(opt.pickle)
|
||||||
|
|
||||||
main(opt)
|
main(opt)
|
||||||
|
|
||||||
"""
|
|
||||||
python3 main.py -d imdb62 -m savlin -A 10 -e 100 -H 32 -c 64 --lr 0.01 -b 250
|
|
||||||
svm: acc=0.915 macrof1=0.915 microf1=0.915
|
|
||||||
network prediction (savlin)
|
|
||||||
acc=90.9000%
|
|
||||||
macro-f1=0.9086
|
|
||||||
micro-f1=0.9090
|
|
||||||
|
|
||||||
end-to-end network prediction (attr)
|
|
||||||
acc=94.6000%
|
|
||||||
macro-f1=0.9458
|
|
||||||
micro-f1=0.9460
|
|
||||||
|
|
||||||
"""
|
|
Loading…
Reference in New Issue