Update main.py to change project name and adjust label_names; modify hidden_layers in Qmlp model
This commit is contained in:
parent
e2dc549be9
commit
5cd09aa696
6
main.py
6
main.py
|
|
@ -5,10 +5,10 @@ from Qfunctions.saveToxlsx import save_to_xlsx as save_to_xlsx
|
|||
|
||||
def main():
|
||||
# 输入元数据文件夹名称
|
||||
projet_name = '20260318Letters'
|
||||
projet_name = '20260319Numbers'
|
||||
# 请在[]内输入每一个分类的名称
|
||||
# label_names 是一个列表里面按顺序包含了小写的‘a'到‘z’
|
||||
label_names = ['a','b']
|
||||
label_names = list(range(10))
|
||||
print(label_names)
|
||||
data = load_data(projet_name, label_names, isDir=False, fileClass='xlsx')
|
||||
X_train, X_test, y_train, y_test, encoder = divSet(
|
||||
|
|
@ -17,7 +17,7 @@ def main():
|
|||
|
||||
model = Qmlp(
|
||||
X_train=X_train, X_test=X_test, y_train=y_train, y_test= y_test,
|
||||
hidden_layers = [64],
|
||||
hidden_layers = [128, 256, 128],
|
||||
dropout_rate=0
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue