avitex/glicko-elixir
1
0
Fork 0
mirror of https://github.com/avitex/elixir-glicko synced 2024-11-24 12:09:58 +00:00
Code Issues Releases Wiki Activity

Improve opts and test

Browse Source
This commit is contained in:
avitex 2017-11-16 13:37:56 +11:00
parent 66133e48d1
commit c80e397073
2 changed files with 22 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
lib/glicko.ex
View File

@ -5,26 +5,28 @@ defmodule Glicko do
GameResult, GameResult,
} }
@epsilon 0.0000001
@default_system_constant 0.8 @default_system_constant 0.8
@default_convergence_tolerance 1.0e-7
@type new_rating_opts_t :: [system_constant: float, convergence_tolerance: float]
@doc """ @doc """
Generate a new Rating from an existing rating and a series of results. Generate a new Rating from an existing rating and a series of results.
""" """
@spec new_rating(player :: Player.t, results :: list(GameResult.t), sys_constant :: float) :: Player.t @spec new_rating(player :: Player.t, results :: list(GameResult.t), opts :: new_rating_opts_t) :: Player.t
def new_rating(player, results, sys_constant \\ @default_system_constant) def new_rating(player, results, opts \\ [])
def new_rating(player = %Player{version: :v1}, results, sys_constant) do def new_rating(player = %Player{version: :v1}, results, opts) do
player player
|> Player.to_v2 |> Player.to_v2
|> do_new_rating(results, sys_constant) |> do_new_rating(results, opts)
|> Player.to_v1 |> Player.to_v1
end end
def new_rating(player = %Player{version: :v2}, results, sys_constant) do def new_rating(player = %Player{version: :v2}, results, opts) do
do_new_rating(player, results, sys_constant) do_new_rating(player, results, opts)
end end
defp do_new_rating(player = %Player{version: :v2}, results, sys_constant) do defp do_new_rating(player = %Player{version: :v2}, results, opts) do
results = Enum.map(results, fn result -> results = Enum.map(results, fn result ->
result = result =
Map.new Map.new
@ -38,7 +40,8 @@ defmodule Glicko do
ctx = ctx =
Map.new Map.new
|> Map.put(:system_constant, sys_constant) |> Map.put(:system_constant, Keyword.get(opts, :system_constant, @default_system_constant))
|> Map.put(:convergence_tolerance, Keyword.get(opts, :convergence_tolerance, @default_convergence_tolerance))
|> Map.put(:results, results) |> Map.put(:results, results)
|> Map.put(:player, player) |> Map.put(:player, player)
|> Map.put(:player_rating_deviation_squared, :math.pow(player.rating_deviation, 2)) |> Map.put(:player_rating_deviation_squared, :math.pow(player.rating_deviation, 2))
@ -134,7 +137,7 @@ defmodule Glicko do
end end
defp iterative_algorithm_body(ctx, a, b, fa, fb) do defp iterative_algorithm_body(ctx, a, b, fa, fb) do
if abs(b - a) > @epsilon do if abs(b - a) > ctx.convergence_tolerance do
c = a + (a - b) * fa / (fb - fa) c = a + (a - b) * fa / (fb - fa)
fc = calc_f(ctx, c) fc = calc_f(ctx, c)
{a, fa} = {a, fa} =

16
test/glicko_test.exs
View File

@ -8,7 +8,7 @@ defmodule GlickoTest do
doctest Glicko doctest Glicko
@player Player.new_v1([rating: 1500, rating_deviation: 200]) @player Player.new_v1([rating: 1500, rating_deviation: 200]) |> Player.to_v2
@results [ @results [
GameResult.new(Player.new_v1([rating: 1400, rating_deviation: 30]), :win), GameResult.new(Player.new_v1([rating: 1400, rating_deviation: 30]), :win),
@ -16,14 +16,16 @@ defmodule GlickoTest do
GameResult.new(Player.new_v1([rating: 1700, rating_deviation: 300]), :loss), GameResult.new(Player.new_v1([rating: 1700, rating_deviation: 300]), :loss),
] ]
@valid_player_rating_after_results 1464.06 @valid_player_rating_after_results 1464.06 |> Player.scale_rating_to(:v2)
@valid_player_rating_deviation_after_results 151.52 @valid_player_rating_deviation_after_results 151.52 |> Player.scale_rating_deviation_to(:v2)
@valid_player_volatility_after_results 0.05999
test "new rating" do test "new rating" do
%Player{rating: new_rating, rating_deviation: new_rating_deviation} = %Player{rating: new_rating, rating_deviation: new_rating_deviation, volatility: new_volatility} =
Glicko.new_rating(@player, @results, 0.5) Glicko.new_rating(@player, @results, [system_constant: 0.5])
assert_in_delta new_rating, @valid_player_rating_after_results, 0.1 assert_in_delta new_rating, @valid_player_rating_after_results, 1.0e-4
assert_in_delta new_rating_deviation, @valid_player_rating_deviation_after_results, 0.1 assert_in_delta new_rating_deviation, @valid_player_rating_deviation_after_results, 1.0e-4
assert_in_delta new_volatility, @valid_player_volatility_after_results, 1.0e-5
end end
end end