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

test

This commit is contained in:
tlk-dsg 2021-09-17 17:46:04 +08:00
parent 60ac409be3
commit 2cbb9327d1
2 changed files with 7 additions and 7 deletions
example/re/standard
predict.pyrun.py

8
example/re/standard/predict.py
View File

@ -4,12 +4,12 @@ import torch
import logging
import hydra
from hydra import utils
from deepke.re_st_tools import Serializer
from deepke.re_st_tools import _serialize_sentence, _convert_tokens_into_index, _add_pos_seq, _handle_relation_data
from deepke.relation_extraction.standard.tools import Serializer
from deepke.relation_extraction.standard.tools import _serialize_sentence, _convert_tokens_into_index, _add_pos_seq, _handle_relation_data
import matplotlib.pyplot as plt
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../")))
from deepke.re_st_utils import load_pkl, load_csv
import deepke.re_st_models as models
from deepke.relation_extraction.standard.utils import load_pkl, load_csv
import deepke.relation_extraction.standard.models as models
logger = logging.getLogger(__name__)

6
example/re/standard/run.py
View File

@ -11,9 +11,9 @@ from torch.utils.tensorboard import SummaryWriter
# self
import sys
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../")))
import deepke.re_st_models as models
from deepke.re_st_tools import preprocess , CustomDataset, collate_fn ,train, validate
from deepke.re_st_utils import manual_seed, load_pkl
import deepke.relation_extraction.standard.models as models
from deepke.relation_extraction.standard.tools import preprocess , CustomDataset, collate_fn ,train, validate
from deepke.relation_extraction.standard.utils import manual_seed, load_pkl
logger = logging.getLogger(__name__)