- 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
)
- For other examples, see Probing Pipeline documentation.