Commit e5603fb6 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

fix examples to rerflect changes in package

parent e23c5e05
import sys
import torch
import os
import itertools
import argparse
from gensim.models import Word2Vec
from corpus import load_corpus
try:
from minibert import *
except:
sys.path.append(os.path.dirname(
os.path.dirname(os.path.realpath(__file__))))
from minibert import *
from .corpus import load_corpus
from minibert import *
from torch.utils.tensorboard import SummaryWriter
from .batch import *
from .evaluation import eval_model, build_eval_batches
......
import os
import torch
from xml.etree import ElementTree
try:
from treetagger import TreeTagger
except:
from .treetagger import TreeTagger
from .treetagger import TreeTagger
from itertools import islice
__all__ = [
......
import torch
from batch import build_one_tensor_batch, build_batches
from .batch import *
def build_eval_batches(sentences, voc2idx, mask_idx, use_cuda):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment