inspirehep / magpie

Deep neural network framework for multi-label text classification

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Input parameters

bigredbug47 opened this issue · comments

Hi all,
Magpie is a wonderful tool, but what about the input parameters when using function, like changes batch size when training or changes embedding size in word2vector.
I found something in config.py file

# word2vec & scaler
EMBEDDING_SIZE = 100

# Cores to use while fitting word2vec vectors
WORD2VEC_WORKERS = 4
MIN_WORD_COUNT = 5
WORD2VEC_CONTEXT = 5

# Models
NN_ARCHITECTURE = 'cnn'

# Training parameters
BATCH_SIZE = 64
EPOCHS = 1

# Number of tokens to save from the abstract, zero padded
SAMPLE_LENGTH = 200

That means if I change something in here it will affect to the magpie function right?
But is there a way that I can pass all necessary parameters directly to magpie function?
Thks.

@bigredbug47 you can pass some of those parameters directly to the function, check the function definitions [here](https://github.com/inspirehep/magpie/blob/master/magpie/main.py#L110-L112). These will overwrite the ones in the file. The rest can be edited in the configuration file.

Does that answer your question?

Thanks, I will take a look at it.