Skip to content

dspy.LabeledFewShot

dspy.LabeledFewShot(k=16)

Bases: Teleprompter

Source code in dspy/teleprompt/vanilla.py 
7
8
def __init__(self, k=16):
    self.k = k

Functions

compile(student, *, trainset, sample=True)

Source code in dspy/teleprompt/vanilla.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
def compile(self, student, *, trainset, sample=True):
    self.student = student.reset_copy()
    self.trainset = trainset

    if len(self.trainset) == 0:
        return self.student

    rng = random.Random(0)

    for predictor in self.student.predictors():
        if sample:
            predictor.demos = rng.sample(self.trainset, min(self.k, len(self.trainset)))
        else:
            predictor.demos = self.trainset[: min(self.k, len(self.trainset))]

    return self.student