add notes
This commit is contained in:
@@ -19,10 +19,12 @@ class Validator:
|
||||
|
||||
self.populate_modules()
|
||||
|
||||
# NOTE: Need to register function here
|
||||
self.dict_corr_map = {
|
||||
"Toxic Queries" : self.Toxic_Queries_correct
|
||||
}
|
||||
|
||||
# NOTE: Need to register function here
|
||||
self.dict_inp_map = {
|
||||
"Toxic Queries" : self.Toxic_Queries_inp_parse
|
||||
}
|
||||
@@ -36,7 +38,7 @@ class Validator:
|
||||
api_key=self.config.api_key
|
||||
)
|
||||
|
||||
|
||||
# NOTE: for every dataset; need one of these
|
||||
def Toxic_Queries_correct(self, inputs: dict, outputs: list, reference_outputs: dict) -> bool:
|
||||
instructions = (
|
||||
"Given an actual answer and an expected answer, determine whether"
|
||||
@@ -63,6 +65,7 @@ class Validator:
|
||||
return response.content.upper() == "CORRECT"
|
||||
|
||||
|
||||
# NOTE: for every dataset; need one of these
|
||||
def Toxic_Queries_inp_parse(self, inp, pipeline:Pipeline):
|
||||
inp = inp["text"]
|
||||
return pipeline.chat(inp, as_raw=True)
|
||||
|
||||
Reference in New Issue
Block a user