diff --git a/main.py b/main.py index 16df139..92a1c81 100644 --- a/main.py +++ b/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 )