Skip to content

Commit adc3c47

Browse files
committed
Create command to train gibberish detector model #2402
Signed-off-by: Jono Yang <jyang@nexb.com>
1 parent a0b3625 commit adc3c47

3 files changed

Lines changed: 27 additions & 0 deletions

File tree

setup-mini.cfg

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,7 @@ console_scripts =
160160
regen-package-docs = packagedcode.regen_package_docs:regen_package_docs
161161
add-required-phrases = licensedcode.required_phrases:add_required_phrases
162162
gen-new-required-phrases-rules = licensedcode.required_phrases:gen_required_phrases_rules
163+
train-gibberish-model = textcode.train_gibberish_model:train_gibberish_model
163164

164165
# These are configurations for ScanCode plugins as setuptools entry points.
165166
# Each plugin entry hast this form:

setup.cfg

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,7 @@ console_scripts =
162162
regen-package-docs = packagedcode.regen_package_docs:regen_package_docs
163163
add-required-phrases = licensedcode.required_phrases:add_required_phrases
164164
gen-new-required-phrases-rules = licensedcode.required_phrases:gen_required_phrases_rules
165+
train-gibberish-model = textcode.train_gibberish_model:train_gibberish_model
165166

166167
# These are configurations for ScanCode plugins as setuptools entry points.
167168
# Each plugin entry hast this form:
Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
#
2+
# Copyright (c) nexB Inc. and others. All rights reserved.
3+
# ScanCode is a trademark of nexB Inc.
4+
# SPDX-License-Identifier: Apache-2.0
5+
# See http://www.apache.org/licenses/LICENSE-2.0 for the license text.
6+
# See https://github.com/nexB/scancode-toolkit for support or download.
7+
# See https://aboutcode.org for more information about nexB OSS projects.
8+
#
9+
10+
import click
11+
12+
from commoncode.cliutils import PluggableCommandLineOption
13+
from textcode.gibberish import Gibberish
14+
15+
@click.command(name='scancode-train-gibberish-model')
16+
@click.help_option('-h', '--help')
17+
def train_gibberish_model(*args, **kwargs,):
18+
"""Train model used by textcode.Gibberish to detect gibberish"""
19+
click.echo('Training gibberish detector model...')
20+
gibberish_detector = Gibberish()
21+
gibberish_detector.train()
22+
23+
24+
if __name__ == '__main__':
25+
train_gibberish_model()

0 commit comments

Comments
 (0)