Skip to content

Latest commit

 

History

History
22 lines (20 loc) · 570 Bytes

usage.md

File metadata and controls

22 lines (20 loc) · 570 Bytes

Usage Examples

  • Basic example:
from probing.pipeline import ProbingPipeline
from pathlib import Path

experiment = ProbingPipeline(
    hf_model_name="DeepPavlov/rubert-base-cased",
    device="cuda:0",
    metric_names=["f1", "accuracy"],
    encoding_batch_size=32,
    classifier_batch_size=32
)

experiment.run(
    probe_task="gapping",
    #path_to_task_file="example.csv",
    verbose=True,
    train_epochs=20
)