Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion parser/base_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,9 @@
import codecs

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from tensorflow.python.client import timeline

from debug.timer import Timer
Expand Down
3 changes: 2 additions & 1 deletion parser/elmo_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,8 @@
import codecs

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.base_network import BaseNetwork
from parser.neural import nn, nonlin, embeddings, recurrent, classifiers
Expand Down
3 changes: 2 additions & 1 deletion parser/graph_outputs.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,8 @@
import time

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.neural import nn
from scripts.chuliu_edmonds import chuliu_edmonds_one_root
Expand Down
3 changes: 2 additions & 1 deletion parser/graph_parser_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,8 @@
import codecs

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.base_network import BaseNetwork
from parser.neural import nn, nonlin, embeddings, recurrent, classifiers
Expand Down
3 changes: 2 additions & 1 deletion parser/neural/classifiers.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,8 @@
from __future__ import print_function

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from . import nn
from . import nonlin
Expand Down
3 changes: 2 additions & 1 deletion parser/neural/embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,8 @@
from __future__ import print_function

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from . import nn
from .classifiers import linear_attention
Expand Down
4 changes: 2 additions & 2 deletions parser/neural/nn.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,15 +60,15 @@ def dropout(inputs, keep_prob, noise_shape=None):

if isinstance(noise_shape, (tuple, list)):
noise_shape = tf.stack(noise_shape)
return tf.nn.dropout(inputs, keep_prob=keep_prob, noise_shape=noise_shape)
return tf.nn.dropout(inputs, rate=1 - keep_prob, noise_shape=noise_shape)

#===============================================================
def unscaled_dropout(inputs, keep_prob, noise_shape=None):
""""""

if isinstance(noise_shape, (tuple, list)):
noise_shape = tf.stack(noise_shape)
return tf.nn.dropout(inputs, keep_prob=keep_prob, noise_shape=noise_shape)*keep_prob
return tf.nn.dropout(inputs, rate=1 - keep_prob, noise_shape=noise_shape)*keep_prob

#===============================================================
def drop_mask(shape, keep_prob):
Expand Down
5 changes: 3 additions & 2 deletions parser/neural/nonlin.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,9 @@
from __future__ import print_function

import numpy as np
import tensorflow as tf

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from tensorflow import identity, tanh, asinh
from tensorflow.python.ops.nn import relu, elu

Expand Down
3 changes: 2 additions & 1 deletion parser/neural/optimizers/adam_optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,8 @@
from __future__ import division
from __future__ import print_function

import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.neural.optimizers.optimizer import Optimizer

Expand Down
3 changes: 2 additions & 1 deletion parser/neural/optimizers/amsgrad_optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,8 @@
from __future__ import division
from __future__ import print_function

import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.neural.optimizers.optimizer import Optimizer

Expand Down
3 changes: 2 additions & 1 deletion parser/neural/optimizers/optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,8 @@
import six

import re
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

#***************************************************************
class Optimizer(object):
Expand Down
3 changes: 2 additions & 1 deletion parser/neural/recurrent.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,8 @@
from __future__ import print_function

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from . import nn
from . import nonlin
Expand Down
3 changes: 2 additions & 1 deletion parser/parser_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,8 @@
import codecs

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.base_network import BaseNetwork
from parser.neural import nn, nonlin, embeddings, recurrent, classifiers
Expand Down
3 changes: 2 additions & 1 deletion parser/structs/buckets/dict_bucket.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,8 @@
from __future__ import print_function

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from .base_bucket import BaseBucket

Expand Down
3 changes: 2 additions & 1 deletion parser/structs/buckets/list_bucket.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,8 @@
from __future__ import print_function

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from .base_bucket import BaseBucket

Expand Down
3 changes: 2 additions & 1 deletion parser/structs/buckets/list_multibucket.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,8 @@
import six

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from .base_multibucket import BaseMultibucket
from .list_bucket import ListBucket
Expand Down
3 changes: 2 additions & 1 deletion parser/structs/conllu_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,8 @@
from collections import Counter

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.structs.buckets import DictMultibucket

Expand Down
3 changes: 2 additions & 1 deletion parser/structs/vocabs/base_vocabs.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,8 @@
from collections import defaultdict as DefaultDict

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

import parser.neural.nn
# id_vocab.root = 0
Expand Down
3 changes: 2 additions & 1 deletion parser/structs/vocabs/feature_vocabs.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,8 @@
from collections import defaultdict as DefaultDict

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.structs.vocabs.base_vocabs import BaseVocab
from . import conllu_vocabs as cv
Expand Down
3 changes: 2 additions & 1 deletion parser/structs/vocabs/index_vocabs.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,8 @@
import six

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from .base_vocabs import BaseVocab
from . import conllu_vocabs as cv
Expand Down
3 changes: 2 additions & 1 deletion parser/structs/vocabs/multivocabs.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,8 @@
from __future__ import print_function

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from .base_vocabs import BaseVocab
from . import conllu_vocabs as cv
Expand Down
3 changes: 2 additions & 1 deletion parser/structs/vocabs/pretrained_vocabs.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,8 @@
from collections import Counter

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.structs.vocabs.base_vocabs import SetVocab
from . import conllu_vocabs as cv
Expand Down
3 changes: 2 additions & 1 deletion parser/structs/vocabs/subtoken_vocabs.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,8 @@
from collections import Counter

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.structs.buckets import ListMultibucket
from .base_vocabs import CountVocab
Expand Down
3 changes: 2 additions & 1 deletion parser/structs/vocabs/token_vocabs.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,8 @@
from collections import Counter

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from parser.structs.vocabs.base_vocabs import CountVocab
from . import conllu_vocabs as cv
Expand Down
3 changes: 2 additions & 1 deletion parser/tagger_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,8 @@
import codecs

import numpy as np
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

from debug.timer import Timer

Expand Down
5 changes: 5 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
tensorflow
pandas
matplotlib
scipy
psutil