TransBoostler/parameters.py at master · MeLLL-UFF/TransBoostler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import os

ROOT_PATH = os.getcwd() + '/'

FASTTEXT = 'fasttext'
WORD2VEC = 'word2vec'

GOOGLE_WORD2VEC = ROOT_PATH + 'resources/word2vec/GoogleNews-vectors-negative300.bin'
GOOGLE_WORD2VEC_SPACY = ROOT_PATH + 'resources/word2vec/spacy'

WIKIPEDIA_FASTTEXT = ROOT_PATH + 'resources/fasttext/wiki.en.vec'
WIKIPEDIA_FASTTEXT_SPACY = ROOT_PATH + 'resources/fasttext/spacy'

WORD_VECTOR_SIMILARITIES = ['cosine','euclidean']

USE_HUNGARIAN_METHOD = False
INCLUDE_TYPES = False

EMBEDDING_DIMENSION = 300
#METHOD = 'CONCATENATE'
METHOD = None

TOP_N = 1

# Validation parameters
N_FOLDS = 3
AMOUNTS = [0.2, 0.4, 0.6, 0.8, 1.0]
AMOUNTS_SMALL = [5, 10, 15, 20, 25]

# BoostSRL parameters
NODESIZE = 2
NUMOFCLAUSES = 8
MAXTREEDEPTH = 3
TREES = 10
SEED = 441773
REFINE = None
ALLOW_SAME_TARGET_MAP = False
SEARCH_PERMUTATION = True
SEARCH_EMPTY = False

# Filenames
TRANSFER_FILENAME = ROOT_PATH + 'boostsrl/transfer.txt'
REFINE_FILENAME = ROOT_PATH + 'boostsrl/refine.txt'
REFINE_REVISION_FILENAME = ROOT_PATH + 'boostsrl/refine-revision.txt'
WILLTHEORIES_FILENAME = ROOT_PATH + 'boostsrl/train/models/WILLtheories/{}_learnedWILLregressionTrees.txt'
SOURCE_TREE_NODES_FILES = 'source_tree_nodes.pkl'
STRUCTURED_TREE_NODES_FILES = 'source_structured_nodes.pkl'

TEST_OUTPUT = ROOT_PATH + 'boostsrl/test/results_{}.db'
TEST_NEGATIVES = ROOT_PATH + 'boostsrl/test/test_neg.txt'
TEST_POSITIVES = ROOT_PATH + 'boostsrl/test/test_pos.txt'

TRAIN_OUTPUT_FILE = ROOT_PATH + 'boostsrl/train_output.txt'
TEST_OUTPUT_FILE = ROOT_PATH + 'boostsrl/test_output.txt'
BACKGROUND_FILE = ROOT_PATH + 'boostsrl/background.txt'

MAX_REVISION_ITERATIONS = 1

#Folders
TRAIN_FOLDER = ROOT_PATH + 'boostsrl/train'
TEST_FOLDER = ROOT_PATH + 'boostsrl/test'
BEST_MODEL_FOLDER = ROOT_PATH + 'boostsrl/best'

TRAIN_FOLDER_FILES = ROOT_PATH + 'boostsrl/train/*'
TEST_FOLDER_FILES = ROOT_PATH + 'boostsrl/test/*'
BEST_MODEL_FOLDER_FILES = ROOT_PATH + 'boostsrl/best/*'