diff --git a/Simulation.py b/Simulation.py
index 70d839b..0fbbcd1 100644
--- a/Simulation.py
+++ b/Simulation.py
@@ -179,11 +179,11 @@ if __name__ == "__main__":
 
     # to check bying
     stock_codes = {
-        #"252670": ['20220602'],
-        "122630": ['20220611', '20220612', '20220613', '20220614'],
+        "252670": ['20220811'],
+        #"122630": ['20220611', '20220612', '20220613', '20220614'],
     }
 
-    method = "rule" # "rule", "ml", "answer"
+    method = "ml" # "rule", "ml", "answer"
     for stock_code in stock_codes:
         simulation = Simulation(RESOURCE_PATH)
 
diff --git a/VitTrainer.py b/VitTrainer.py
index bea1205..b3afed1 100755
--- a/VitTrainer.py
+++ b/VitTrainer.py
@@ -236,5 +236,5 @@ if __name__ == "__main__":
     stock_code = "252670"
     vitTrainer = VitTrainer(RESOURCE_PATH)
 
-    train_ds, val_ds = vitTrainer.getData(stock_code, sDate="20220809", eDate="20220812")
+    train_ds, val_ds = vitTrainer.getData(stock_code, sDate="20220701", eDate="20220731")
     vitTrainer.train(train_ds, val_ds, model_path)