Update training model experiment with autokeras

[Yullia]

Description

Please include a summary of the change and which is fixed.

• Add A
• Fix B

How has this been tested?

Please describe the tests that you ran to verify your changes.

• [ ] UnitTest
• [ ] Benchmark