-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.py
29 lines (23 loc) · 894 Bytes
/
config.py
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
#------------UNIVERSAL------------
#Antall forskjellige ord
VOCABULARY_SIZE = 5000
#antall ord pr tweet
TWEET_LENGTH = 50
#------------TRAINING------------
#Hvis False starter trener main en ny modell, mens hvis LOAD WEIGHT er True laster
# main inn vektene gitt i WEIGHT_FILE og fortsetter treningen derfra.
LOAD_WEIGHT = False
WEIGHT_FILE = "weights/last_huge.h5"
GENERAL_TWEETS_LOCATION = "data/training/"
LEARNING_RATE=1e-3
#Antall epoker for å trene modellen
EPOCHS = 100
#Hvor mange tweets som trenes på parallellt.
BATCH_SIZE = 256
#------------MODEL------------
#der hvor glove embedding filen skal hentes fra
EMBEDDING_FILE_LOCATION = "C:\Cogito\glove\glove.twitter.27B.100d.txt"
#------------APPLIKASJON------------
# Velger hvilke vekter man skal bruke i applikasjonen
LOAD_WEIGHT_PREDICT = "weights/best_trump_huge4.h5"
LOAD_CASE_WEIGHT_PREDICT = "weights/bestCaseLoss2.h5"