diff --git a/csv_merge.py b/csv_merge.py index 0d2cc40..58ee723 100644 --- a/csv_merge.py +++ b/csv_merge.py @@ -20,8 +20,8 @@ def process(directory): def merge(mainDF:pd.DataFrame, candDF:pd.DataFrame, outname): #add new columns to main dataframe colNum = len(mainDF.columns) - mainDF.insert(colNum, "probability", 0) - mainDF.insert(colNum+1, "label", 0) + mainDF.insert(colNum, "probability", float(0)) + mainDF.insert(colNum+1, "label", float(0)) #iterate over candidates for props in candDF.itertuples(index=False):