Transformers documentation

Công cụ tiện ích cho pipelines

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v5.22.1).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Công cụ tiện ích cho pipelines

Trang này liệt kê tất cả các hàm tiện ích mà thư viện cung cấp cho pipelines.

Phần lớn những điều đó chỉ hữu ích nếu bạn đang nghiên cứu mã nguồn của các mô hình trong thư viện.

Xử lý đối số

lớp transformers.pipelines.ArgumentHandler

< >

( )

Giao diện cơ bản để xử lý các đối số cho mỗi Pipeline.

lớp transformers.pipelines.ZeroShotClassificationArgumentHandler

< >

( )

Xử lý các đối số cho zero-shot trong phân loại văn bản bằng cách biến mỗi nhãn có thể thành một cặp premise/hypothesis cho NLI.

Định dạng dữ liệu

lớp transformers.PipelineDataFormat

< >

( output_path: str | Noneinput_path: str | Nonecolumn: str | Noneoverwrite: bool = False )

Tham số

  • output_path (str) — Nơi lưu trữ dữ liệu đầu ra.
  • input_path (str) — Nơi tìm dữ liệu đầu vào.
  • column (str) — Cột cần đọc.
  • overwrite (bool, tùy chọn, mặc định là False) — Có ghi đè lên output_path hay không.

Lớp cơ sở cho tất cả các định dạng dữ liệu được pipeline hỗ trợ, cả khi đọc và ghi. Các định dạng dữ liệu được hỗ trợ hiện tại bao gồm:

  • JSON
  • CSV
  • stdin/stdout (ống dẫn)

PipelineDataFormat cũng bao gồm một số tiện ích để làm việc với nhiều cột, chẳng hạn như ánh xạ từ các cột của datasets sang các tham số keyword của pipelines thông qua định dạng dataset_kwarg_1=dataset_column_1.

from_str

< >

( format: stroutput_path: str | Noneinput_path: str | Nonecolumn: str | Noneoverwrite = False ) → PipelineDataFormat

Tham số

  • format (str) — Định dạng của pipeline mong muốn. Các giá trị chấp nhận được là "json", "csv" hoặc "pipe".
  • output_path (str, tùy chọn) — Nơi lưu trữ dữ liệu đầu ra.
  • input_path (str, tùy chọn) — Nơi tìm dữ liệu đầu vào.
  • column (str, tùy chọn) — Cột cần đọc.
  • overwrite (bool, tùy chọn, mặc định là False) — Có ghi đè lên output_path hay không.

Trả về

PipelineDataFormat

Định dạng dữ liệu phù hợp.

Tạo một instance của subclass đúng của PipelineDataFormat tùy thuộc vào format.

lưu

< >

( data: dict | list[dict] )

Tham số

  • data (dict hoặc danh sách dict) — Dữ liệu cần lưu trữ.

Lưu đối tượng dữ liệu được cung cấp với biểu diễn cho PipelineDataFormat hiện tại.

save_binary

< >

( data: dict | list[dict] ) → str

Tham số

  • data (dict hoặc danh sách dict) — Dữ liệu cần lưu trữ.

Trả về

str

Đường dẫn nơi dữ liệu đã được lưu trữ.

Lưu đối tượng dữ liệu được cung cấp dưới dạng dữ liệu nhị phân định dạng pickle trên đĩa.

lớp transformers.CsvPipelineDataFormat

< >

( output_path: str | Noneinput_path: str | Nonecolumn: str | Noneoverwrite = False )

Tham số

  • output_path (str) — Nơi lưu trữ dữ liệu đầu ra.
  • input_path (str) — Nơi tìm dữ liệu đầu vào.
  • column (str) — Cột cần đọc.
  • overwrite (bool, tùy chọn, mặc định là False) — Có ghi đè lên output_path hay không.

Hỗ trợ cho các pipeline sử dụng định dạng dữ liệu CSV.

lưu

< >

( data: list[dict] )

Tham số

  • data (list[dict]) — Dữ liệu cần lưu trữ.

Lưu đối tượng dữ liệu được cung cấp với biểu diễn cho PipelineDataFormat hiện tại.

lớp transformers.JsonPipelineDataFormat

< >

( output_path: str | Noneinput_path: str | Nonecolumn: str | Noneoverwrite = False )

Tham số

  • output_path (str) — Nơi lưu trữ dữ liệu đầu ra.
  • input_path (str) — Nơi tìm dữ liệu đầu vào.
  • column (str) — Cột cần đọc.
  • overwrite (bool, tùy chọn, mặc định là False) — Có ghi đè lên output_path hay không.

Hỗ trợ cho các pipeline sử dụng định dạng tệp JSON.

lưu

< >

( data: dict )

Tham số

  • data (dict) — Dữ liệu cần lưu trữ.

Lưu đối tượng dữ liệu được cung cấp vào một tệp json.

lớp transformers.PipedPipelineDataFormat

< >

( output_path: str | Noneinput_path: str | Nonecolumn: str | Noneoverwrite: bool = False )

Tham số

  • output_path (str) — Nơi lưu trữ dữ liệu đầu ra.
  • input_path (str) — Nơi tìm dữ liệu đầu vào.
  • column (str) — Cột cần đọc.
  • overwrite (bool, tùy chọn, mặc định là False) — Có ghi đè lên output_path hay không.

Đọc dữ liệu từ đầu vào dạng ống (piped input) đến tiến trình python. Đối với dữ liệu nhiều cột, các cột nên được phân tách bằng

Nếu các cột được cung cấp, thì đầu ra sẽ là một từ điển có dạng {column_x: value_x}

lưu

< >

( data: dict )

Tham số

  • data (dict) — Dữ liệu cần lưu trữ.

In dữ liệu.

Công cụ tiện ích

lớp transformers.pipelines.PipelineException

< >

( task: strmodel: strreason: str )

Tham số

  • task (str) — Nhiệm vụ của pipeline.
  • model (str) — Mô hình được pipeline sử dụng.
  • reason (str) — Thông báo lỗi cần hiển thị.

Được Pipeline phát sinh khi xử lý call.

Update on GitHub