durantdelor/vision_stock_pro

1#!/usr/bin/env python32"""3Script de test pour vérifier le chargement des données sur Hugging Face4"""5 6import os7import pandas as pd8 9def test_data_loading():10    """Test du chargement des données"""11    print("🔍 Test du chargement des données...")12    13    # Vérifier l'existence du dossier data14    if not os.path.exists('data'):15        print("❌ Dossier 'data' non trouvé")16        return False17    18    print("✅ Dossier 'data' trouvé")19    20    # Lister les fichiers CSV21    csv_files = [f for f in os.listdir('data') if f.endswith('.csv')]22    print(f"📁 Fichiers CSV trouvés: {len(csv_files)}")23    for file in csv_files:24        print(f"  - {file}")25    26    # Tester le chargement de chaque fichier27    csv_files_dict = {28        'mayor1_csv': 'data/sample_data_clean.csv',29        'laitbroli_1kg': 'data/laitbroli_1kg_clean.csv',30        'may_arm_1kg': 'data/may_arm_1kg_clean.csv',31        'may_arm_5kg': 'data/may_arm_5kg_clean.csv',32        'couche_softcqre_T4': 'data/couche_softcqre_T4_clean.csv',33        'papierhygsita': 'data/papierhygsita_clean.csv',34        'parleG': 'data/parleG_clean.csv'35    }36    37    for key, file_path in csv_files_dict.items():38        print(f"\n🔍 Test de {key}: {file_path}")39        40        if not os.path.exists(file_path):41            print(f"❌ Fichier non trouvé: {file_path}")42            continue43        44        try:45            # Charger le fichier CSV46            df = pd.read_csv(file_path, encoding='utf-8', sep=',', on_bad_lines='skip')47            print(f"✅ Fichier chargé: {len(df)} lignes, {len(df.columns)} colonnes")48            print(f"   Colonnes: {list(df.columns)}")49            50            # Vérifier les types de données51            print(f"   Types: {df.dtypes.to_dict()}")52            53            # Vérifier les premières lignes54            print(f"   Premières lignes:")55            print(df.head(2).to_string())56            57        except Exception as e:58            print(f"❌ Erreur lors du chargement: {e}")59    60    return True61 62if __name__ == "__main__":63    test_data_loading()64