.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/dqn_farm1_rlberry.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        Click :ref:`here <sphx_glr_download_auto_examples_dqn_farm1_rlberry.py>`
        to download the full example code

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_dqn_farm1_rlberry.py:


rlberry DQN on Farm1 from farm-gym-games
========================================

This example use farm-gym-games and rlberry, please install these libraries before using.

pip install git+https://github.com/rlberry-py/rlberry.git
pip install git+https://github.com/farm-gym/farm-gym-games

.. GENERATED FROM PYTHON SOURCE LINES 10-90

.. code-block:: default



    import farmgym_games
    import numpy as np
    from rlberry.envs import gym_make

    from rlberry.agents.torch import DQNAgent
    from rlberry.manager import AgentManager, evaluate_agents, plot_writer_data
    from rlberry.agents.torch.utils.training import model_factory_from_env

    import pandas as pd
    import seaborn as sns
    import time
    import matplotlib.pyplot as plt


    actions_txt = [
        "doing nothing",
        "1L of water",
        "5L of water",
        "harvesting",
        "sow some seeds",
        "scatter fertilizer",
        "scatter herbicide",
        "scatter pesticide",
        "remove weeds by hand",
    ]


    env_ctor, env_kwargs = gym_make, {"id": "Farm1-v0"}

    if __name__ == "__main__":
        manager = AgentManager(
            DQNAgent,
            (env_ctor, env_kwargs),
            agent_name="DQNAgent",
            init_kwargs=dict(
                learning_rate=9e-5,
            ),
            fit_budget=5e5,
            eval_kwargs=dict(eval_horizon=365),
            n_fit=1,
            parallelization="process",
            mp_context="spawn",
            enable_tensorboard=True,
            seed=42,
        )

        init_time = time.time()
        manager.fit()
        print("training time in s is ", time.time() - init_time)
        fig, ax = plt.subplots(figsize=(12, 6))
        data = plot_writer_data(manager, tag="episode_rewards", smooth_weight=0.8, ax=ax)  # smoothing tensorboard-style

        fig.savefig("dqn_regret.pdf")

        agent = manager.agent_handlers[0]  # select the agent from the manager
        env = gym_make("Farm1-v0")

        rew = 0
        while rew < 1:
            rew = 0
            obs = env.reset()
            episode = pd.DataFrame()
            for day in range(365):
                action = agent.policy(obs)
                obs, reward, is_done, _ = env.step(action)
                episode = pd.concat(
                    [episode, pd.DataFrame({"action": [actions_txt[action]], "reward": [reward]})], ignore_index=True
                )
                rew = rew + reward
                if is_done:
                    print("Plant is Dead")
                    break

            print(rew)

        fig, ax = plt.subplots(figsize=(12, 6))
        sns.countplot(data=episode, x="action", order=episode["action"].value_counts().index, ax=ax)
        fig.savefig("dqn_barplot.pdf")


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** ( 0 minutes  0.000 seconds)


.. _sphx_glr_download_auto_examples_dqn_farm1_rlberry.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example


    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: dqn_farm1_rlberry.py <dqn_farm1_rlberry.py>`

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: dqn_farm1_rlberry.ipynb <dqn_farm1_rlberry.ipynb>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_