flaredetector: dont check pcov

This commit is contained in:
2024-05-13 11:33:10 +02:00
parent 24979f97cd
commit 277a1eb473
-5
View File
@@ -33,15 +33,10 @@ def calculateFitForFlare(lightcurve, peakIndex):
if(not isValidFlarePoint(flareData, newPeakIndex)): if(not isValidFlarePoint(flareData, newPeakIndex)):
raise Exception("Flare not valid: not enough datapoints above mean") raise Exception("Flare not valid: not enough datapoints above mean")
if(PeakMinus1 > 0.01 and PeakMinus1 < PeakMinus2): if(PeakMinus1 > 0.01 and PeakMinus1 < PeakMinus2):
if(Peak > 0.025):
print(lightcurve.time[peakIndex], lightcurve.flux[peakIndex])
print(lightcurve.flux[peakIndex], lightcurve.flux[peakIndex-1], lightcurve.flux[peakIndex-2])
raise Exception(f"Flare not valid: peak-2 > peak-1, {PeakMinus2} > {PeakMinus1}") raise Exception(f"Flare not valid: peak-2 > peak-1, {PeakMinus2} > {PeakMinus1}")
# TODO: allow peak-1 to be around/lower than mean too if theres enough points after # TODO: allow peak-1 to be around/lower than mean too if theres enough points after
popt, pcov = curve_fit(lambda x, l, w: GaussExpo(x, Peak, l, w), flareFitDataPoints, flareData) popt, pcov = curve_fit(lambda x, l, w: GaussExpo(x, Peak, l, w), flareFitDataPoints, flareData)
pcovDiag = np.diag(pcov) pcovDiag = np.diag(pcov)
if(pcovDiag[0] > 1 and pcovDiag[1] > 1):
raise Exception("Flare not valid: could not properly fit")
fit = GaussExpo(flareFitDataPoints, Peak, *popt) fit = GaussExpo(flareFitDataPoints, Peak, *popt)
flareFitDataPoints = flareFitDataPoints + peakIndex flareFitDataPoints = flareFitDataPoints + peakIndex
return Peak+1, flareFitDataPoints, fit+1 return Peak+1, flareFitDataPoints, fit+1