Tianshou/tianshou/data/__init__.py

from tianshou.data.batch import Batch
from tianshou.data.utils.converter import to_numpy, to_torch, \
    to_torch_as
from tianshou.data.utils.segtree import SegmentTree
from tianshou.data.buffer import ReplayBuffer, \
    ListReplayBuffer, PrioritizedReplayBuffer
from tianshou.data.collector import Collector

__all__ = [
    'Batch',
    'to_numpy',
    'to_torch',
    'to_torch_as',
    'SegmentTree',
    'ReplayBuffer',
    'ListReplayBuffer',
    'PrioritizedReplayBuffer',
    'Collector',
]
env and data 2020-03-11 09:09:56 +08:00			`from tianshou.data.batch import Batch`
Improve PER (#159) - use segment tree to rewrite the previous PrioReplayBuffer code, add the test - enable all Q-learning algorithms to use PER 2020-08-06 10:26:24 +08:00			`from tianshou.data.utils.converter import to_numpy, to_torch, \`
nstep all (fix #51) 2020-06-03 13:59:47 +08:00			`to_torch_as`
Improve PER (#159) - use segment tree to rewrite the previous PrioReplayBuffer code, add the test - enable all Q-learning algorithms to use PER 2020-08-06 10:26:24 +08:00			`from tianshou.data.utils.segtree import SegmentTree`
add ListReplayBuffer 2020-03-28 15:14:41 +08:00			`from tianshou.data.buffer import ReplayBuffer, \`
			`ListReplayBuffer, PrioritizedReplayBuffer`
half of collector 2020-03-12 22:20:33 +08:00			`from tianshou.data.collector import Collector`
env and data 2020-03-11 09:09:56 +08:00
maybe finished collector? 2020-03-13 17:49:22 +08:00			`__all__ = [`
			`'Batch',`
Robust conversion from/to numpy/pytorch (#63) * Enable to convert Batch data back to torch. * Add torch converter to collector. * Fix * Move to_numpy/to_torch convert in dedicated utils.py. * Use to_numpy/to_torch to convert arrays. * fix lint * fix * Add unit test to check Batch from/to numpy. * Fix Batch over Batch. Co-authored-by: Alexis Duburcq <alexis.duburcq@wandercraft.eu> 2020-05-29 14:45:21 +02:00			`'to_numpy',`
			`'to_torch',`
nstep all (fix #51) 2020-06-03 13:59:47 +08:00			`'to_torch_as',`
Improve PER (#159) - use segment tree to rewrite the previous PrioReplayBuffer code, add the test - enable all Q-learning algorithms to use PER 2020-08-06 10:26:24 +08:00			`'SegmentTree',`
maybe finished collector? 2020-03-13 17:49:22 +08:00			`'ReplayBuffer',`
add ListReplayBuffer 2020-03-28 15:14:41 +08:00			`'ListReplayBuffer',`
maybe finished collector? 2020-03-13 17:49:22 +08:00			`'PrioritizedReplayBuffer',`
Improve PER (#159) - use segment tree to rewrite the previous PrioReplayBuffer code, add the test - enable all Q-learning algorithms to use PER 2020-08-06 10:26:24 +08:00			`'Collector',`
maybe finished collector? 2020-03-13 17:49:22 +08:00			`]`