# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: Python 3
#     language: python
#     name: python3
# ---

# +
# Do not delete this cell. It ensures that you can do the imports,
# load datasets etc. in the same fashion as in any Python script
# in the project template.


import sys
sys.path.insert(0, '../..')
from bld.project_paths import project_paths_join as ppj
from bld.project_paths import project_paths as pp



# +
import numpy as np
import pickle
# %config Completer.use_jedi = False


# -

with open(ppj("OUT_DATA", f"grid_3_agents.pickle"), "rb") as f:
    all_grids = pickle.load(f)

with open(ppj("OUT_DATA", f"all_super_stars_3_agents.pickle"), "rb") as f:
    super_stars = pickle.load(f)

with open(ppj("OUT_ANALYSIS", f"array_no_deviation_simulations_3_agents.pickle"), "rb") as f:
    super_star_price_data = pickle.load(f)

super_star_price_data.mean()

# +
    alpha_grid = list(
        np.linspace(
            0.025,
            0.25,
            100,
        )
    )
    beta_grid = list(
        np.linspace(
            1e-8,
            2e-05,
            100,
        )
    )

    beta_index = beta_grid.index( 6.157575757575758e-07)
    alpha_index = alpha_grid[::-1].index(0.02954545454545455)


# -

np.flipud(np.mean(all_grids['avg_price'], axis=0))[alpha_index, beta_index]

StarcoderdataJupyterScriptsDedupFiltered