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

Simplify example

Browse source 
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
This commit is contained in:
Andras Schmelczer 2022-04-10 14:43:21 +02:00
parent 76c9f4a0cd
commit 0877d34b8b
9 changed files with 18 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

8
example/train.ipynb
View file

@ -29,8 +29,7 @@
"from good_ai.utilities.language import is_english, predict_language\n",
"from good_ai import save_model, set_default_config, LargeFile\n",
"\n",
"from preprocess import preprocess\n",
"from config import model_key"
"from preprocess import preprocess"
]
},
{
@ -48,7 +47,8 @@
"source": [
"PREFIX = \"domain-\"\n",
"DATASET_KEY = \"data\"\n",
"MAX_FILE_COUNT = 5"
"MAX_FILE_COUNT = 5\n",
"MODEL_KEY = \"small-domain-prediction-v2\""
]
},
{
@ -774,7 +774,7 @@
}
],
"source": [
"save_model(classifier, key=model_key, keep_last_n=1)"
"save_model(classifier, key=MODEL_KEY, keep_last_n=1)"
]
}
],