File size: 1,044 Bytes
8360ec7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import random
from openfactcheck.core.solver import Solver
from openfactcheck.core.state import FactCheckerState
from openfactcheck.core.solver import StandardTaskSolver, Solver
@Solver.register("fake_claim_extractor", "response", "claims")
class FakeClaimExtractor(StandardTaskSolver):
def __init__(self, args):
super().__init__(args)
self.max_claims = args.get("max_claims", 5)
self.min_claims = args.get("min_claims", 2)
def __call__(self, state: FactCheckerState, *args, **kwargs):
response = state.get(self.input_name)
response_len = len(response)
num_claims = random.randint(min(self.min_claims, response_len), min(self.min_claims, response_len))
cut_pont = list(range(response_len))
random.shuffle(cut_pont)
cut_pont = sorted(cut_pont[:num_claims + 1])
claims = []
for i in range(len(cut_pont) - 1):
claims.append(response[cut_pont[i]:cut_pont[i + 1]])
state.set(self.output_name, claims)
return True, state
|