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

pip

This commit is contained in:
tlk-dsg 2021-07-02 20:35:42 +08:00
parent 94d23febef
commit 685ee23ef9
2 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tools/main.py
View File

@ -11,11 +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 models
from preprocess import preprocess
from dataset import CustomDataset, collate_fn
from trainer import train, validate
from utils import manual_seed, load_pkl
import deepke.models as models
from deepke.tools import preprocess , CustomDataset, collate_fn ,train, validate
from deepke.utils import manual_seed, load_pkl
logger = logging.getLogger(__name__)

8
tools/predict.py
View File

@ -4,12 +4,12 @@ import torch
import logging
import hydra
from hydra import utils
from serializer import Serializer
from preprocess import _serialize_sentence, _convert_tokens_into_index, _add_pos_seq, _handle_relation_data
from deepke.tools import Serializer
from deepke.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 utils import load_pkl, load_csv
import models
from deepke.utils import load_pkl, load_csv
import deepke.models as models
logger = logging.getLogger(__name__)