ModelSmith-AI / backend /utils /validators.py
ACA050's picture
Upload 79 files
a309487 verified
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