andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Imrpove examples

Browse source
This commit is contained in:
Andras Schmelczer 2022-05-26 13:33:27 +02:00
parent c44b47c084
commit 7192ba064a
7 changed files with 308 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

4
examples/complex/main_batch.py
View file

@ -4,10 +4,10 @@ import json
from random import shuffle
from devtools import debug
from predict_domain import predict_domain
from great_ai import process_batch
from predict_domain import predict_domain
if __name__ == "__main__":
with open(".cache/data-1/s2-corpus-323.json") as f:
raw = json.load(f)

6
examples/complex/predict_domain.py
View file

@ -1,12 +1,12 @@
import re
from typing import Dict, Iterable, List
from preprocess import preprocess
from great_ai import log_argument, log_metric, use_model
from great_ai.utilities.clean import clean
from pydantic import BaseModel
from sklearn.pipeline import Pipeline
from great_ai import log_argument, log_metric, use_model
from great_ai.utilities.clean import clean
from preprocess import preprocess
class DomainPrediction(BaseModel):