Update main.py to change project name and adjust label_names; modify hidden_layers in Qmlp model

This commit is contained in:
qyh1510@gmail.com 2026-03-28 21:28:20 +08:00
parent e2dc549be9
commit 5cd09aa696
1 changed files with 3 additions and 3 deletions

View File

@ -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
)