bacancydataprophets/Metal_Defect_Detection

1import pandas as pd2 3class Compare:4    def comparison(self, df, df1):5        final = df.merge(df1, on='Image/File Name')6        final.drop_duplicates(subset=['Confidence Score'], keep='first', inplace=True)7        final = final.reset_index().drop(columns=['index'])8        9        fnl_cls = []10        fnl_cnf = []11        for i in range(0, len(final['Confidence Score'])):12            if final['Confidence Score'][i] > final['Conf score'][i]:13                fnl_cls.append(final['Detected Class'][i])14                fnl_cnf.append(final['Confidence Score'][i])15            else:16                fnl_cls.append(final['Detected class by cls'][i])17                fnl_cnf.append(final['Conf score'][i])18        19        final['fnl_cls'] = fnl_cls20        final['fnl_pred'] = fnl_cnf21        final = final.drop(columns=['Detected Class', 'Confidence Score', 'Detected class by cls', 'Conf score'])22        return final23