thu-ml · Trinkle23897 · Apr 24, 2022 · Mar 8, 2022 · Mar 8, 2022 · Mar 8, 2022
diff --git a/docs/_static/js/benchmark.js b/docs/_static/js/benchmark.js
@@ -1,4 +1,4 @@
-var envs = [
+var mujoco_envs = [
     "Ant-v3",
     "HalfCheetah-v3",
     "Hopper-v3",
@@ -9,12 +9,24 @@ var envs = [
     "Swimmer-v3",
     "Walker2d-v3",
 ];
-function showEnv(elem) {
-    var selectEnv = elem.value || envs[0];
+
+var atari_envs = [
+    "PongNoFrameskip-v4",
+    "BreakoutNoFrameskip-v4",
+    "EnduroNoFrameskip-v4",
+    "QbertNoFrameskip-v4",
+    "MsPacmanNoFrameskip-v4",
+    "SeaquestNoFrameskip-v4",
+    "SpaceInvadersNoFrameskip-v4",
+];
+
+function showMujocoEnv(elem) {
+    var selectEnv = elem.value || mujoco_envs[0];
     var dataSource = {
         $schema: "https://vega.github.io/schema/vega-lite/v5.json",
         data: {
-            url: "/en/master/_static/js/mujoco/benchmark/" + selectEnv + "/result.json"
+            // url: "/_static/js/mujoco/benchmark/" + selectEnv + "/result.json"
+            url: "/en/master/_static/js/atari/benchmark/" + selectEnv + "/result.json"
         },
         mark: "line",
         height: 400,
@@ -58,10 +70,67 @@ function showEnv(elem) {
     };
     vegaEmbed("#vis-mujoco", dataSource);
 }
+
+function showAtariEnv(elem) {
+    var selectEnv = elem.value || atari_envs[0];
+    var dataSource = {
+        $schema: "https://vega.github.io/schema/vega-lite/v5.json",
+        data: {
+            // url: "/_static/js/atari/benchmark/" + selectEnv + "/result.json"
+            url: "/en/master/_static/js/atari/benchmark/" + selectEnv + "/result.json"
+        },
+        mark: "line",
+        height: 400,
+        width: 800,
+        params: [{name: "Range", value: 10000000, bind: {input: "range", min: 10000, max: 10000000}}],
+        transform: [
+            {calculate: "datum.rew - datum.rew_std", as: "rew_std0"},
+            {calculate: "datum.rew + datum.rew_std", as: "rew_std1"},
+            {calculate: "datum.rew + ' ± ' + datum.rew_std", as: "tooltip_str"},
+            {filter: "datum.env_step <= Range"},
+        ],
+        encoding: {
+            color: {"field": "Agent", "type": "nominal"},
+            x: {field: "env_step", type: "quantitative", title: "Env step"},
+        },
+        layer: [{
+            "encoding": {
+                "opacity": {"value": 0.3},
+                "y": {
+                    "title": "Return",
+                    "field": "rew_std0",
+                    "type": "quantitative",
+                },
+                "y2": {"field": "rew_std1"},
+                tooltip: [
+                    {field: "env_step", type: "quantitative", title: "Env step"},
+                    {field: "Agent", type: "nominal"},
+                    {field: "tooltip_str", type: "nominal", title: "Return"},
+                ]
+            },
+            "mark": "area"
+        }, {
+            "encoding": {
+                "y": {
+                    "field": "rew",
+                    "type": "quantitative"
+                }
+            },
+            "mark": "line"
+        }]
+    };
+    vegaEmbed("#vis-atari", dataSource);
+}
+
 $(document).ready(function() {
-    var envSelect = $("#env-mujoco");
-    if (envSelect.length) {
-        $.each(envs, function(idx, env) {envSelect.append($("<option></option>").val(env).html(env));})
-        showEnv(envSelect);
+    var envMujocoSelect = $("#env-mujoco");
+    if (envMujocoSelect.length) {
+        $.each(mujoco_envs, function(idx, env) {envMujocoSelect.append($("<option></option>").val(env).html(env));})
+        showMujocoEnv(envMujocoSelect);
+    }
+    var envAtariSelect = $("#env-atari");
+    if (envAtariSelect.length) {
+        $.each(atari_envs, function(idx, env) {envAtariSelect.append($("<option></option>").val(env).html(env));})
+        showAtariEnv(envAtariSelect);
     }
 });
diff --git a/docs/tutorials/benchmark.rst b/docs/tutorials/benchmark.rst
@@ -12,7 +12,7 @@ Every experiment is conducted under 10 random seeds for 1-10M steps. Please refe
 .. raw:: html
 
     <center>
-        <select id="env-mujoco" onchange="showEnv(this)"></select>
+        <select id="env-mujoco" onchange="showMujocoEnv(this)"></select>
         <br>
         <div id="vis-mujoco"></div>
         <br>
@@ -95,3 +95,13 @@ Atari Benchmark
 ---------------
 
 Please refer to https://github.com/thu-ml/tianshou/tree/master/examples/atari
+
+.. raw:: html
+
+    <center>
+        <select id="env-atari" onchange="showAtariEnv(this)"></select>
+        <br>
+        <div id="vis-atari"></div>
+        <br>
+    </center>
+
diff --git a/examples/atari/benchmark/BreakoutNoFrameskip-v4/result.json b/examples/atari/benchmark/BreakoutNoFrameskip-v4/result.json
diff --git a/examples/atari/benchmark/EnduroNoFrameskip-v4/result.json b/examples/atari/benchmark/EnduroNoFrameskip-v4/result.json
diff --git a/examples/atari/benchmark/MsPacmanNoFrameskip-v4/result.json b/examples/atari/benchmark/MsPacmanNoFrameskip-v4/result.json
diff --git a/examples/atari/benchmark/PongNoFrameskip-v4/result.json b/examples/atari/benchmark/PongNoFrameskip-v4/result.json
diff --git a/examples/atari/benchmark/QbertNoFrameskip-v4/result.json b/examples/atari/benchmark/QbertNoFrameskip-v4/result.json
diff --git a/examples/atari/benchmark/SeaquestNoFrameskip-v4/result.json b/examples/atari/benchmark/SeaquestNoFrameskip-v4/result.json
diff --git a/examples/atari/benchmark/SpaceInvadersNoFrameskip-v4/result.json b/examples/atari/benchmark/SpaceInvadersNoFrameskip-v4/result.json