50efccc10d7dd32caf913b7da3059551dd7739d6,deepchem/splits/__init__.py,StratifiedSplitter,__generate_required_hits,#StratifiedSplitter#Any#Any#,96

Before Change

        required_hit_dict = {}
        totalCount = len(y_df.index)
        for col in y_df:
            NaN_count = y_df[col].isnull().sum()
            notNaN = totalCount - NaN_count
            requiredNotNaN = frac_train * notNaN
            required_hit_dict[colIndex] = requiredNotNaN
            colIndex += 1
        return required_hit_dict

    def __generate_required_index(self, y_df, required_hit_dict):

After Change

    def __generate_required_hits(self, w, frac_train):
        required_hits = (w != 0).sum(0)//returns list of per column sum of non zero elements
        for colHits in required_hits:
          colHits = int(frac_train * colHits)
        return required_hits

    def __generate_required_index(self, w, required_hit_list):
        colIndex = 0

Instances