def METHOD_NAME(args, dataset_name):
    data_loader = None
    if dataset_name == "20news":
        data_loader = data.raw_data_loader.news_20.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name == "agnews":
        data_loader = data.raw_data_loader.AGNews.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name == "cnn_dailymail":
        data_loader = data.raw_data_loader.CNN_Dailymail.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name == "cornell_movie_dialogue":
        data_loader = (
            data.raw_data_loader.CornellMovieDialogue.data_loader.RawDataLoader(
                args.data_dir
            )
        )
    elif dataset_name == "semeval_2010_task8":
        data_loader = data.raw_data_loader.SemEval2010Task8.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name == "sentiment140":
        data_loader = data.raw_data_loader.Sentiment140.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name == "squad_1.1":
        data_loader = data.raw_data_loader.SQuAD_1_1.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name == "sst_2":
        data_loader = data.raw_data_loader.SST_2.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name == "w_nut":
        data_loader = data.raw_data_loader.W_NUT.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name == "wikiner":
        data_loader = data.raw_data_loader.wikiner.data_loader.RawDataLoader(
            args.data_dir
        )
    elif dataset_name.startswith("wmt"):
        data_loader = data.raw_data_loader.WMT.data_loader.RawDataLoader(
            args.data_dir.split(",")
        )
    else:
        raise Exception("No such dataset!!")
    logger.info("Start loading data from files")
    start = time.time()
    data_loader.load_data()
    end = time.time()
    logger.info("Finish loading data from files, it takes %f sec" % (end - start))
    logger.info("Start generating h5 data file")
    start = time.time()
    data_loader.generate_h5_file(args.h5_file_path)
    end = time.time()
    logger.info("Finish generating h5 data file, it takes %f sec" % (end - start))