├── .gitignore ├── LICENSE ├── README.md ├── benchmark ├── acrobot │ ├── agents │ │ ├── random_shooting.py │ │ └── random_shooting_for_real_system.py │ ├── data │ │ ├── linear │ │ │ ├── X_test.csv │ │ │ └── X_train.csv │ │ ├── metadata.json │ │ ├── random │ │ │ ├── X_test.csv │ │ │ ├── X_train.csv │ │ │ └── trace.csv │ │ └── real_traces_random.csv │ ├── env.py │ ├── generate_static_trace.py │ ├── problem.py │ ├── reward_function.py │ └── submissions │ │ ├── arlin_sigma │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darmdn │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darmdn_det │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darmdn_single │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darnn_det │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darnn_sigma │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── dmdn │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── gpytorch │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── real_nvp │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── starting_kit │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ └── vae │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py ├── acrobot_sincos │ ├── agents │ │ ├── random_shooting.py │ │ └── random_shooting_for_real_system.py │ ├── data │ │ ├── linear │ │ │ ├── X_test.csv │ │ │ └── X_train.csv │ │ ├── metadata.json │ │ ├── random │ │ │ ├── X_test.csv │ │ │ ├── X_train.csv │ │ │ └── trace.csv │ │ └── real_traces_random.csv │ ├── env.py │ ├── generate_static_trace.py │ ├── problem.py │ ├── reward_function.py │ └── submissions │ │ ├── arlin_sigma │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darmdn │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darmdn_det │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darmdn_single │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darmdn_single_det │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darnn_det │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── darnn_sigma │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── dmdn │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── dmdn_single │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── dmdn_single_det │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── gpytorch │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── real_nvp │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ ├── starting_kit │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py │ │ └── vae │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py ├── inverted_pendulum │ ├── agents │ │ ├── random_shooting.py │ │ └── random_shooting_for_real_system.py │ ├── data │ │ ├── X_test.csv │ │ ├── X_train.csv │ │ ├── metadata.json │ │ └── trace.csv │ ├── env.py │ ├── generate_static_trace.py │ ├── problem.py │ ├── reward_function.py │ └── submissions │ │ └── starting_kit │ │ ├── generative_regressor.py │ │ └── ts_feature_extractor.py └── inverted_pendulum_reset │ ├── agents │ ├── random_shooting.py │ └── random_shooting_for_real_system.py │ ├── data │ ├── X_test.csv │ ├── X_train.csv │ ├── metadata.json │ └── trace.csv │ ├── env.py │ ├── generate_static_trace.py │ ├── problem.py │ ├── reward_function.py │ └── submissions │ └── starting_kit │ ├── generative_regressor.py │ └── ts_feature_extractor.py ├── environment.yml └── mbrl-tools ├── README.md ├── mbrltools ├── __init__.py ├── data_processing.py ├── model_based_rl.py ├── model_env.py └── pytorch_utils.py ├── setup.py └── tests ├── conftest.py ├── small_acrobot ├── agents │ └── random_shooting.py ├── data │ └── metadata.json ├── env.py ├── problem.py ├── reward_function.py └── submissions │ ├── dummy_kit │ ├── generative_regressor.py │ └── ts_feature_extractor.py │ └── starting_kit │ ├── generative_regressor.py │ └── ts_feature_extractor.py ├── test_model_based_rl.py ├── test_model_env.py └── test_pytorch_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/acrobot/agents/random_shooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/agents/random_shooting.py -------------------------------------------------------------------------------- /benchmark/acrobot/agents/random_shooting_for_real_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/agents/random_shooting_for_real_system.py -------------------------------------------------------------------------------- /benchmark/acrobot/data/linear/X_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/data/linear/X_test.csv -------------------------------------------------------------------------------- /benchmark/acrobot/data/linear/X_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/data/linear/X_train.csv -------------------------------------------------------------------------------- /benchmark/acrobot/data/metadata.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/data/metadata.json -------------------------------------------------------------------------------- /benchmark/acrobot/data/random/X_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/data/random/X_test.csv -------------------------------------------------------------------------------- /benchmark/acrobot/data/random/X_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/data/random/X_train.csv -------------------------------------------------------------------------------- /benchmark/acrobot/data/random/trace.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/data/random/trace.csv -------------------------------------------------------------------------------- /benchmark/acrobot/data/real_traces_random.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/data/real_traces_random.csv -------------------------------------------------------------------------------- /benchmark/acrobot/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/env.py -------------------------------------------------------------------------------- /benchmark/acrobot/generate_static_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/generate_static_trace.py -------------------------------------------------------------------------------- /benchmark/acrobot/problem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/problem.py -------------------------------------------------------------------------------- /benchmark/acrobot/reward_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/reward_function.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/arlin_sigma/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/arlin_sigma/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/arlin_sigma/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/arlin_sigma/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darmdn/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darmdn/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darmdn/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darmdn/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darmdn_det/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darmdn_det/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darmdn_det/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darmdn_det/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darmdn_single/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darmdn_single/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darmdn_single/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darmdn_single/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darnn_det/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darnn_det/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darnn_det/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darnn_det/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darnn_sigma/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darnn_sigma/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/darnn_sigma/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/darnn_sigma/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/dmdn/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/dmdn/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/dmdn/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/dmdn/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/gpytorch/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/gpytorch/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/gpytorch/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/gpytorch/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/real_nvp/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/real_nvp/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/real_nvp/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/real_nvp/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/starting_kit/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/starting_kit/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/starting_kit/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/starting_kit/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/vae/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/vae/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot/submissions/vae/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot/submissions/vae/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/agents/random_shooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/agents/random_shooting.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/agents/random_shooting_for_real_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/agents/random_shooting_for_real_system.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/data/linear/X_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/data/linear/X_test.csv -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/data/linear/X_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/data/linear/X_train.csv -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/data/metadata.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/data/metadata.json -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/data/random/X_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/data/random/X_test.csv -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/data/random/X_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/data/random/X_train.csv -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/data/random/trace.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/data/random/trace.csv -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/data/real_traces_random.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/data/real_traces_random.csv -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/env.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/generate_static_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/generate_static_trace.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/problem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/problem.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/reward_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/reward_function.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/arlin_sigma/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/arlin_sigma/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/arlin_sigma/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/arlin_sigma/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darmdn/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darmdn/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darmdn/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darmdn/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darmdn_det/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darmdn_det/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darmdn_det/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darmdn_det/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darmdn_single/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darmdn_single/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darmdn_single/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darmdn_single/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darmdn_single_det/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darmdn_single_det/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darmdn_single_det/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darmdn_single_det/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darnn_det/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darnn_det/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darnn_det/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darnn_det/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darnn_sigma/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darnn_sigma/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/darnn_sigma/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/darnn_sigma/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/dmdn/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/dmdn/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/dmdn/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/dmdn/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/dmdn_single/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/dmdn_single/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/dmdn_single/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/dmdn_single/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/dmdn_single_det/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/dmdn_single_det/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/dmdn_single_det/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/dmdn_single_det/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/gpytorch/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/gpytorch/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/gpytorch/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/gpytorch/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/real_nvp/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/real_nvp/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/real_nvp/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/real_nvp/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/starting_kit/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/starting_kit/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/starting_kit/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/starting_kit/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/vae/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/vae/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/acrobot_sincos/submissions/vae/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/acrobot_sincos/submissions/vae/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/agents/random_shooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/agents/random_shooting.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/agents/random_shooting_for_real_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/agents/random_shooting_for_real_system.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/data/X_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/data/X_test.csv -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/data/X_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/data/X_train.csv -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/data/metadata.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/data/metadata.json -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/data/trace.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/data/trace.csv -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/env.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/generate_static_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/generate_static_trace.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/problem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/problem.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/reward_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/reward_function.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/submissions/starting_kit/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/submissions/starting_kit/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum/submissions/starting_kit/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum/submissions/starting_kit/ts_feature_extractor.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/agents/random_shooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/agents/random_shooting.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/agents/random_shooting_for_real_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/agents/random_shooting_for_real_system.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/data/X_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/data/X_test.csv -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/data/X_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/data/X_train.csv -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/data/metadata.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/data/metadata.json -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/data/trace.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/data/trace.csv -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/env.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/generate_static_trace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/generate_static_trace.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/problem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/problem.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/reward_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/reward_function.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/submissions/starting_kit/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/submissions/starting_kit/generative_regressor.py -------------------------------------------------------------------------------- /benchmark/inverted_pendulum_reset/submissions/starting_kit/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/benchmark/inverted_pendulum_reset/submissions/starting_kit/ts_feature_extractor.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/environment.yml -------------------------------------------------------------------------------- /mbrl-tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/README.md -------------------------------------------------------------------------------- /mbrl-tools/mbrltools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mbrl-tools/mbrltools/data_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/mbrltools/data_processing.py -------------------------------------------------------------------------------- /mbrl-tools/mbrltools/model_based_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/mbrltools/model_based_rl.py -------------------------------------------------------------------------------- /mbrl-tools/mbrltools/model_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/mbrltools/model_env.py -------------------------------------------------------------------------------- /mbrl-tools/mbrltools/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/mbrltools/pytorch_utils.py -------------------------------------------------------------------------------- /mbrl-tools/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/setup.py -------------------------------------------------------------------------------- /mbrl-tools/tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/conftest.py -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/agents/random_shooting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/agents/random_shooting.py -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/data/metadata.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/data/metadata.json -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/env.py -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/problem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/problem.py -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/reward_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/reward_function.py -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/submissions/dummy_kit/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/submissions/dummy_kit/generative_regressor.py -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/submissions/dummy_kit/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/submissions/dummy_kit/ts_feature_extractor.py -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/submissions/starting_kit/generative_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/submissions/starting_kit/generative_regressor.py -------------------------------------------------------------------------------- /mbrl-tools/tests/small_acrobot/submissions/starting_kit/ts_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/small_acrobot/submissions/starting_kit/ts_feature_extractor.py -------------------------------------------------------------------------------- /mbrl-tools/tests/test_model_based_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/test_model_based_rl.py -------------------------------------------------------------------------------- /mbrl-tools/tests/test_model_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/test_model_env.py -------------------------------------------------------------------------------- /mbrl-tools/tests/test_pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ramp-kits/rl_simulator/HEAD/mbrl-tools/tests/test_pytorch_utils.py --------------------------------------------------------------------------------