felixm/ML4T
1
0
Fork 0
Code Releases Activity

Fix mistake in previous solution and finish report for project 1.

Browse Source
This commit is contained in:
Felix Martin
2020-08-07 15:55:12 -04:00
parent 27675adf59
commit a11cc99a88
6 changed files with 74 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
martingale/martingale.py
View File

@@ -90,7 +90,7 @@ def experiment_1_figure_2(number_runs=1000):
configure_plot()
runs = np.array([run_experiment() for _ in range(number_runs)])
winnings_mean = runs.mean(axis=0)
winnings_std = winnings_mean.std()
winnings_std = runs.std(axis=0)
plt.plot(winnings_mean, linewidth=0.7)
plt_std_setting = {'ls': '-', 'color': 'blue', 'linewidth': 0.3}
@@ -103,7 +103,7 @@ def experiment_1_figure_2(number_runs=1000):
def experiment_1_figure_3(runs, figurename='figure_3.png'):
configure_plot()
winnings_median = np.median(a=runs, axis=0)
winnings_std = winnings_median.std()
winnings_std = runs.std(axis=0)
plt.plot(winnings_median, linewidth=0.7)
plt_std_setting = {'ls': '-', 'color': 'blue', 'linewidth': 0.3}
plt.plot(winnings_median + winnings_std, **plt_std_setting)
@@ -115,7 +115,7 @@ def experiment_2_figure_4(number_runs=1000):
runs = np.array([run_experiment(initial_credit=256)
for _ in range(number_runs)])
winnings_mean = runs.mean(axis=0)
winnings_std = winnings_mean.std()
winnings_std = runs.std(axis=0)
plt.plot(winnings_mean, linewidth=0.7)
plt_std_setting = {'ls': '-', 'color': 'blue', 'linewidth': 0.3}
plt.plot(winnings_mean + winnings_std, **plt_std_setting)