newbie
/
Deeplearning
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@ -5,10 +5,10 @@ from Qfunctions.saveToxlsx import save_to_xlsx as save_to_xlsx
def main(): def main():
# 输入元数据文件夹名称 # 输入元数据文件夹名称
projet_name = '20260318Letters' projet_name = '20260319Numbers'
# 请在[]内输入每一个分类的名称 # 请在[]内输入每一个分类的名称
# label_names 是一个列表里面按顺序包含了小写的‘a'到‘z’ # label_names 是一个列表里面按顺序包含了小写的‘a'到‘z’
label_names = ['a','b'] label_names = list(range(10))
print(label_names) print(label_names)
data = load_data(projet_name, label_names, isDir=False, fileClass='xlsx') data = load_data(projet_name, label_names, isDir=False, fileClass='xlsx')
X_train, X_test, y_train, y_test, encoder = divSet( X_train, X_test, y_train, y_test, encoder = divSet(
@ -17,7 +17,7 @@ def main():
model = Qmlp( model = Qmlp(
X_train=X_train, X_test=X_test, y_train=y_train, y_test= y_test, 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 dropout_rate=0
) )