Repository navigation
What's New
- MLP Hyperparameter Optimization:
--optimizeflag in train_reranker.py uses RandomizedSearchCV to search over architecture, learning rate, alpha, activation, and max_iter - Per-Domain Threshold Tuning:
--optimize-thresholdsflag finds optimal mlp_uncertainty_low/high values per domain, maximizing accuracy + coverage - New training params:
alpha,learning_rate_init,activationexposed in bothMLPReranker.train()andarf.trainer.train_reranker()public API - Config schema validation: New MLP fields (mlp_alpha, mlp_learning_rate_init, mlp_hidden_layer_sizes, mlp_activation, mlp_max_iter) in config_schema.py
Install
pip install advanced-rag-framework==0.2.2