SkyID
/
deepke
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

test

This commit is contained in:
tlk-dsg 2021-09-16 16:17:57 +08:00
parent d8f64dabfb
commit 60ac409be3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
example/ae/standard/predict.py
View File

@ -4,8 +4,8 @@ import torch
import logging
import hydra
from hydra import utils
from deepke.ae_re_tools import Serializer
from deepke.ae_re_tools import _serialize_sentence, _convert_tokens_into_index, _add_pos_seq, _handle_attribute_data
from deepke.ae_st_tools import Serializer
from deepke.ae_st_tools import _serialize_sentence, _convert_tokens_into_index, _add_pos_seq, _handle_attribute_data
import matplotlib.pyplot as plt
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../")))
from deepke.ae_st_utils import load_pkl, load_csv