INNER CODE UNIT · Python

classify

xbeat/Machine-Learning · organize.py:238

def classify(filename):
    name = filename.replace(".md", "").lower()
    for folder in FOLDER_ORDER:
        patterns = CATEGORIES[folder]
        for pat in patterns:
            if re.search(pat, name, re.IGNORECASE):
                return folder
    return "Misc"


def main():
    files = [f for f in os.listdir(BASE) if f.endswith(".md") and os.path.isfile(os.path.join(BASE, f))]

    # Count per category first
    counts = {folder: [] for folder in FOLDER_ORDER}
    counts["Misc"] = []
    for f in files:
        cat = classify(f)

View source record →