open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

number filtering curriculum (#333)

Browse source
This commit is contained in:
Zafir Stojanovski 2025-03-11 23:56:06 +01:00 • committed by GitHub
parent f14662e213
commit aa6ccf1946
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 81 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/algorithmic/__init__.py
View file

@ -22,7 +22,7 @@ from .jugs import JugsConfig, JugsDataset
from .letter_counting import LetterCountingConfig, LetterCountingCurriculum, LetterCountingDataset
from .letter_jumble import LetterJumbleConfig, LetterJumbleCurriculum, LetterJumbleDataset
from .manipulate_matrix import ManipulateMatrixConfig, ManipulateMatrixCurriculum, ManipulateMatrixDataset
from .number_filtering import NumberFilteringConfig, NumberFilteringDataset
from .number_filtering import NumberFilteringConfig, NumberFilteringCurriculum, NumberFilteringDataset
from .number_sorting import NumberSortingConfig, NumberSortingCurriculum, NumberSortingDataset
from .palindrome_generation import PalindromeConfig, PalindromeCurriculum, PalindromeDataset
from .palindrome_partitioning import (
@ -73,6 +73,7 @@ __all__ = [
"LetterJumbleCurriculum",
"NumberFilteringConfig",
"NumberFilteringDataset",
"NumberFilteringCurriculum",
"NumberSortingConfig",
"NumberSortingDataset",
"NumberSortingCurriculum",

47
reasoning_gym/algorithmic/number_filtering.py
View file

@ -4,6 +4,7 @@ from dataclasses import dataclass
from random import Random
from typing import Optional
from ..coaching import AttributeType, BaseCurriculum, RangeAttributeDefinition
from ..factory import ProceduralDataset, register_dataset
@ -94,8 +95,52 @@ class NumberFilteringDataset(ProceduralDataset):
"filter_value": filter_str,
"operation": f"{keep_remove}_{larger_smaller}",
"result": result_strs,
"difficulty": {
"numbers": len(numbers),
"decimals": (self.config.min_decimals, self.config.max_decimals),
"value": (self.config.min_value, self.config.max_value),
},
},
}
register_dataset("number_filtering", NumberFilteringDataset, NumberFilteringConfig)
class NumberFilteringCurriculum(BaseCurriculum):
def __init__(self):
super().__init__(NumberFilteringCurriculum.__name__, NumberFilteringConfig)
# Define attributes
self._define_attributes(
RangeAttributeDefinition(
name="numbers",
levels=[10, 100, 500, 1000],
default_level=1,
description="How many numbers to sort",
attr_type=AttributeType.APPEND,
min_value=2,
lower_field_name="min_numbers",
upper_field_name="max_numbers",
),
RangeAttributeDefinition(
name="decimals",
levels=[0, 2, 4, 6],
default_level=1,
description="Number of decimal places",
attr_type=AttributeType.APPEND,
min_value=0,
lower_field_name="min_decimals",
upper_field_name="max_decimals",
),
RangeAttributeDefinition(
name="value",
levels=[-10_000, 10_000],
default_level=1,
description="Range of numbers to sort",
attr_type=AttributeType.APPEND,
min_value=-10_000,
lower_field_name="min_value",
upper_field_name="max_value",
),
)
register_dataset("number_filtering", NumberFilteringDataset, NumberFilteringConfig, NumberFilteringCurriculum)