Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Mlflow+EarlyStopping #75

Merged
merged 22 commits into from
Aug 6, 2024
Merged

Mlflow+EarlyStopping #75

merged 22 commits into from
Aug 6, 2024

Conversation

Anya497
Copy link
Collaborator

@Anya497 Anya497 commented Jul 23, 2024

  • Create EarlyStopping and add it to the training pipeline.
  • Separate two types of learning with study's URI in config file. If you want to train model with concrete hyperparameters you need to specify URI of optuna's study that contains information about the best trial with hyperparameters you interested in. Also you need to add to config model's weights URI which correspond to chosen model's parameters. Otherwise optuna's hyperparameters tuning will be started.
  • Add validation type to config. Move number of servers to configuration of validation with SVMs.
  • Speed up loss validation with batches.
  • Add logging with MLFlow.

@Anya497 Anya497 requested review from gsvgit, emnigma and Parzival-05 and removed request for gsvgit and emnigma July 23, 2024 07:25
@Anya497 Anya497 requested a review from Parzival-05 July 24, 2024 13:08
AIAgent/common/classes.py Show resolved Hide resolved
AIAgent/common/game.py Show resolved Hide resolved
AIAgent/ml/play_game.py Show resolved Hide resolved
AIAgent/ml/play_game.py Show resolved Hide resolved
AIAgent/run_training.py Show resolved Hide resolved
AIAgent/ml/play_game.py Outdated Show resolved Hide resolved
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants