diff --git a/fink_science/__init__.py b/fink_science/__init__.py index b1a1c157..dd98dce4 100644 --- a/fink_science/__init__.py +++ b/fink_science/__init__.py @@ -12,4 +12,4 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -__version__ = "4.9.0" +__version__ = "4.10.0" diff --git a/fink_science/random_forest_snia/processor.py b/fink_science/random_forest_snia/processor.py index ac74cafb..6fbddead 100644 --- a/fink_science/random_forest_snia/processor.py +++ b/fink_science/random_forest_snia/processor.py @@ -183,6 +183,10 @@ def rfscore_sigmoid_full(jd, fid, magpsf, sigmapsf, cdsxmatch, ndethist, model=N # Make predictions probabilities = clf.predict_proba(test_features) + # pIa = 0.0 for objects that do not + # have both features non-zero. + probabilities[~flag] = [1.0, 0.0] + # Take only probabilities to be Ia to_return = np.zeros(len(jd), dtype=float) to_return[mask] = probabilities.T[1]