Skip to content

Commit

Permalink
tidy
Browse files Browse the repository at this point in the history
  • Loading branch information
RobinL committed Apr 6, 2020
1 parent 2343735 commit 073a83b
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions splink/iterate.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,18 +41,12 @@ def iterate(
DataFrame: A spark dataframe including a match probability column
"""

# df_gammas.persist()

num_iterations = settings["max_iterations"]
for i in range(num_iterations):
df_e = run_expectation_step(
df_gammas, params, settings, spark, compute_ll=compute_ll
)

# To align to the parameters in the iteration chart, LL must be computed against
# pi and lambda pre-maximisation
# compute_log_likelihood()

run_maximisation_step(df_e, params, spark)

logger.info(f"Iteration {i} complete")
Expand All @@ -68,6 +62,4 @@ def iterate(
df_gammas, params, settings, spark, compute_ll=compute_ll
)

# df_gammas.unpersist()
return df_e

0 comments on commit 073a83b

Please sign in to comment.