# -*- coding: utf-8 -*-
"""
Created on Mon Nov 12 14:23:12 2018

@author: cm
"""

import os
import sys
pwd = os.path.dirname(os.path.abspath(__file__))
sys.path.append(pwd)

class Hyperparamters:
    # 参数设置
    n_epoch = 20     
    print_step = 100
    train_rate = 0.999   
    # 模型参数设置
    batch_size = 16#128#256#8       
    batch_size_predict = 1
    learning_rate = 5e-5 
    
    # Optimization parameters
    num_train_epochs = 20
    warmup_proportion = 0.1
    use_tpu = None
    do_lower_case = True
    num_labels = 3
    num_filters = 128

    # CNN parameters
    sequence_length = 60#
    conv1_left = 100    
    # TextCNN parameters
    filter_sizes = [2,3,4,5,6,7]
    embedding_size = 384
    keep_prob = 0.5
        
    ## bert path
    name = 'albert_small_zh_google'
    bert_path = os.path.join(pwd,name)
    data_dir = os.path.join(pwd,'data')
    vocab_file = os.path.join(pwd,name,'vocab_chinese.txt')   
    #  
    init_checkpoint = os.path.join(pwd,name,'albert_model.ckpt')
    saved_model_path = os.path.join(pwd,'model')    
    
    
    
    
    
    
    
    


    
    