fir erf

droplet_growth/fit.py

 import numpy as np
 import matplotlib.pyplot as plt
 from scipy.optimize import curve_fit
-from scipy.special import lambertw
+from scipy.special import erf
 import pandas as pd
 import seaborn as sns
 
@@ -265,6 +265,8 @@ def sigmoid(x, a, b):
 def hill(x, n, K):
     return x ** n/(x ** n + K)
 
+def erf_fun(x, a, b):
+    return ((1 + erf(a * (x - b))) / 2)
 
 def fit_sigmoid(probs, ax, fun=sigmoid, fit_name='sigmoid', p0=(2., -2.)):
     probs = probs.copy()