backend/core/problem_inference.py

class ProblemInference:
    def infer(self, dataset_info, target_column):
        if dataset_info.get("possible_nlp"):
            return "nlp"
        
        if target_column:
            return "classification" if dataset_info.get("class_distribution") else "regression"
        
        return "unknown"