12_ATBNN_Demo / paras.py
JHao2830's picture
Upload 9 files
556849d
raw
history blame
278 Bytes
attrib_feature = ['Current','Voltage','Environment_Temperature','Cell_Temperature','SOC']
attrib_label = ['SOH']
max_len = 200
input_dim = len(attrib_feature)
output_dim = 1
hidden_dim = 64
num_layers = 2
num_heads = 4
lr = 1e-3
max_seq_len = 200
batch_size = 5
C_rated = 1.1