LICENSE.md
README.md
pyproject.toml
requirements.txt
src/__init__.py
src/atgen/__init__.py
src/atgen.egg-info/PKG-INFO
src/atgen.egg-info/SOURCES.txt
src/atgen.egg-info/dependency_links.txt
src/atgen.egg-info/entry_points.txt
src/atgen.egg-info/requires.txt
src/atgen.egg-info/top_level.txt
src/atgen/labellers/__init__.py
src/atgen/labellers/base_labeller.py
src/atgen/labellers/custom_llm_labeller.py
src/atgen/labellers/get_labeller.py
src/atgen/labellers/golden_labeller.py
src/atgen/labellers/human_labeller.py
src/atgen/labellers/labelling_utils.py
src/atgen/labellers/api_labellers/__init__.py
src/atgen/labellers/api_labellers/anthropic_labeller.py
src/atgen/labellers/api_labellers/openai_labeller.py
src/atgen/metrics/__init__.py
src/atgen/metrics/bart_score.py
src/atgen/metrics/big_bench_hard_metric.py
src/atgen/metrics/compute_metrics.py
src/atgen/metrics/metrics.py
src/atgen/metrics/supported_models_and_metrics.py
src/atgen/run_scripts/__init__.py
src/atgen/run_scripts/run_active_learning.py
src/atgen/strategies/__init__.py
src/atgen/strategies/base_strategy.py
src/atgen/strategies/bleuvar.py
src/atgen/strategies/dual.py
src/atgen/strategies/get_strategy.py
src/atgen/strategies/graph_cut.py
src/atgen/strategies/hadas.py
src/atgen/strategies/huds.py
src/atgen/strategies/idds.py
src/atgen/strategies/nsp.py
src/atgen/strategies/random_strategy.py
src/atgen/strategies/te_delfy.py
src/atgen/strategies/unieval/__init__.py
src/atgen/strategies/unieval/evaluator.py
src/atgen/strategies/unieval/scorer.py
src/atgen/strategies/unieval/utils.py
src/atgen/utils/__init__.py
src/atgen/utils/check_performance_metrics.py
src/atgen/utils/check_required_performance.py
src/atgen/utils/combine_results.py
src/atgen/utils/constants.py
src/atgen/utils/find_response_token_ids_in_text.py
src/atgen/utils/generate.py
src/atgen/utils/get_embeddings.py
src/atgen/utils/get_initial_labeled_data.py
src/atgen/utils/get_last_workdir.py
src/atgen/utils/installers.py
src/atgen/utils/load_model_tokenizer.py
src/atgen/utils/main_decorator.py
src/atgen/utils/prepare_model_for_training.py
src/atgen/utils/resolvers.py
src/atgen/utils/save_labeled_data.py
src/atgen/utils/save_log_iter_results.py
src/atgen/utils/training_utils.py
src/atgen/utils/validate_and_fill_config.py
src/atgen/utils/data/__init__.py
src/atgen/utils/data/get_output_field.py
src/atgen/utils/data/get_preprocess_function.py
src/atgen/utils/data/load_data.py
src/atgen/utils/data/maybe_get_few_shot_examples.py
src/atgen/utils/data/prepare_conversational_data.py
src/atgen/visualize/__init__.py
src/atgen/visualize/plot_line.py
test/test_benchmark.py