Python ClassBalancedStoragePolicyの例

コード例 #1

ファイル: test_plugins.py プロジェクト: bbeatrix/avalanche

 def test_replay_balanced_memory(self):
     mem_size = 25
     policies = [None,
                 ExperienceBalancedStoragePolicy({}, mem_size=mem_size),
                 ClassBalancedStoragePolicy({}, mem_size=mem_size)]
     for policy in policies:
         self._test_replay_balanced_memory(policy, mem_size)

コード例 #2

ファイル: test_replay.py プロジェクト: stjordanis/avalanche

    def test_balancing(self):
        p1 = ExperienceBalancedStoragePolicy({}, 100, adaptive_size=True)
        p2 = ClassBalancedStoragePolicy({}, 100, adaptive_size=True)

        for policy in [p1, p2]:
            self.assert_balancing(policy)

コード例 #3

ファイル: test_replay.py プロジェクト: vishalbelsare/avalanche

 def setUp(self) -> None:
     self.memory = {}
     self.policy = ClassBalancedStoragePolicy(self.memory, mem_size=4)