blob: 51dfec113658fc08918d9120940c831176b08bb5 (
plain)
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
|
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
# import collections
# import math
# import modeling
# import optimization
import tokenization
# import six
# import tensorflow as tf
# import os
# Global variables
# GPU number, default: -1, means not used
CUDA_VISIBLE_DEVICES = "2"
# Questions to be trained/predicted
questions = ['Communication Service Name', 'Max Number of UEs', 'Data Rate Downlink', 'Latency', 'Data Rate Uplink', 'Resource Sharing Level', 'Mobility', 'Area']
# Configuration file
FLAGS_bert_config_file = '/home/run/uncased_L-12_H-768_A-12/bert_config.json'
FLAGS_vocab_file = '/home/run/uncased_L-12_H-768_A-12/vocab.txt'
FLAGS_init_checkpoint_squad = '/home/run/uncased_L-12_H-768_A-12/bert_model.ckpt'
max_seq_length = 512
tokenizer_ch = tokenization.FullTokenizer(vocab_file=FLAGS_vocab_file, do_lower_case=True)
|
