explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 09:56:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Don't fix random seeds on import

Browse Source
This commit is contained in:
Johannes Dollinger 2018-02-13 12:42:23 +01:00
parent c63e99da8a
commit bf94c13382
3 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/cli/evaluate.py
View File

@ -3,8 +3,6 @@ from __future__ import unicode_literals, division, print_function
import plac import plac
from timeit import default_timer as timer from timeit import default_timer as timer
import random
import numpy.random
from ..gold import GoldCorpus from ..gold import GoldCorpus
from ..util import prints from ..util import prints
@ -12,10 +10,6 @@ from .. import util
from .. import displacy from .. import displacy
random.seed(0)
numpy.random.seed(0)
@plac.annotations( @plac.annotations(
model=("model name or path", "positional", None, str), model=("model name or path", "positional", None, str),
data_path=("location of JSON-formatted evaluation data", "positional", data_path=("location of JSON-formatted evaluation data", "positional",
@ -31,6 +25,8 @@ def evaluate(model, data_path, gpu_id=-1, gold_preproc=False, displacy_path=None
Evaluate a model. To render a sample of parses in a HTML file, set an Evaluate a model. To render a sample of parses in a HTML file, set an
output directory as the displacy_path argument. output directory as the displacy_path argument.
""" """
util.fix_random_seed()
if gpu_id >= 0: if gpu_id >= 0:
util.use_gpu(gpu_id) util.use_gpu(gpu_id)
util.set_env_log(False) util.set_env_log(False)

6
spacy/cli/train.py
View File

@ -6,8 +6,6 @@ from pathlib import Path
import tqdm import tqdm
from thinc.neural._classes.model import Model from thinc.neural._classes.model import Model
from timeit import default_timer as timer from timeit import default_timer as timer
import random
import numpy.random
from ..gold import GoldCorpus, minibatch from ..gold import GoldCorpus, minibatch
from ..util import prints from ..util import prints
@ -16,9 +14,6 @@ from .. import about
from .. import displacy from .. import displacy
from ..compat import json_dumps from ..compat import json_dumps
random.seed(0)
numpy.random.seed(0)
@plac.annotations( @plac.annotations(
lang=("model language", "positional", None, str), lang=("model language", "positional", None, str),
@ -45,6 +40,7 @@ def train(lang, output_dir, train_data, dev_data, n_iter=30, n_sents=0,
""" """
Train a model. Expects data in spaCy's JSON format. Train a model. Expects data in spaCy's JSON format.
""" """
util.fix_random_seed()
util.set_env_log(True) util.set_env_log(True)
n_sents = n_sents or None n_sents = n_sents or None
output_path = util.ensure_path(output_dir) output_path = util.ensure_path(output_dir)

6
spacy/util.py
View File

@ -17,6 +17,7 @@ from thinc.neural._classes.model import Model
import functools import functools
import cytoolz import cytoolz
import itertools import itertools
import numpy as np
from .symbols import ORTH from .symbols import ORTH
from .compat import cupy, CudaStream, path2str, basestring_, input_, unicode_ from .compat import cupy, CudaStream, path2str, basestring_, input_, unicode_
@ -623,3 +624,8 @@ def use_gpu(gpu_id):
Model.ops = CupyOps() Model.ops = CupyOps()
Model.Ops = CupyOps Model.Ops = CupyOps
return device return device
def fix_random_seed(seed=0):
random.seed(0)
np.random.seed(0)