URL占比计算器 - 基于URL字符占比的文本特征提取
输入列名 | 说明 |
|---|---|
texts | 待处理的文本列,要求元素类型为字符串 |
占比结果列,元素为浮点数,表示URL字符的占比
下面的代码展示了如何使用 daft 运行算子计算文本中URL字符的占比。
from __future__ import annotations import os import daft from daft import col from daft.las.functions.text.url_ratio_calculator import UrlRatioCalculator from daft.las.functions.udf import las_udf if __name__ == "__main__": if os.getenv("DAFT_RUNNER", "native") == "ray": import logging import ray def configure_logging(): logging.basicConfig( level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S.%s".format(), ) logging.getLogger("tracing.span").setLevel(logging.WARNING) logging.getLogger("daft_io.stats").setLevel(logging.WARNING) logging.getLogger("DaftStatisticsManager").setLevel(logging.WARNING) logging.getLogger("DaftFlotillaScheduler").setLevel(logging.WARNING) logging.getLogger("DaftFlotillaDispatcher").setLevel(logging.WARNING) ray.init(dashboard_host="0.0.0.0", runtime_env={"worker_process_setup_hook": configure_logging}) daft.context.set_runner_ray() daft.set_execution_config(actor_udf_ready_timeout=600) daft.set_execution_config(min_cpu_per_task=0) samples = { "text": [ "Hello world", "Check out https://example.com for more info", "Visit http://test.com and https://demo.org", "No URLs here, just text", "Mixed content: https://short.com and some text", ] } ds = daft.from_pydict(samples) ds = ds.with_column( "url_ratio", las_udf( UrlRatioCalculator, construct_args={}, )(col("text")), ) ds.show() # ╭────────────────────────────────┬────────────────────╮ # │ text ┆ url_ratio │ # │ --- ┆ --- │ # │ String ┆ Float64 │ # ╞════════════════════════════════╪════════════════════╡ # │ Hello world ┆ 0 │ # ├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤ # │ Check out https://example.com… ┆ 0.4418604651162791 │ # ├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤ # │ Visit http://test.com and htt… ┆ 0.7380952380952381 │ # ├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤ # │ No URLs here, just text ┆ 0 │ # ├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤ # │ Mixed content: https://short.… ┆ 0.3695652173913043 │ # ╰────────────────────────────────┴────────────────────╯