File size: 468 Bytes
a309487
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import pandas as pd

class DataValidator:
    def validate_dataframe(self, df: pd.DataFrame, target_column: str):
        if df.empty:
            raise ValueError("Uploaded dataset is empty.")

        if target_column not in df.columns:
            raise ValueError(f"Target column '{target_column}' not found in dataset.")

        if df.shape[1] < 2:
            raise ValueError("Dataset must have at least 2 columns.")

        return True