crash.software
Projects
Pull Requests
Issues
Builds
SIEVA
Code
Files
Commits
Branches
Tags
Pull Requests
Code Comments
Code Compare
Issues
List
Boards
Milestones
Builds
Statistics
Contributions
Source Lines
Child Projects
Projects
STRLCPY
SIEVA
Files
🤬
Sign In
8f7a8e2d
ROOT
/
AI_Engine
/
core
/
model.py
Search
History
122 lines
|
UTF-8
|
6 KB
Blame
Outline
2
4
8
No wrap
Soft wrap
Outline
np
pd
utils
os
joblib
simple_preprocess
train_supervised
load_model
accuracy_score
recall_score
f1_score
classification_report
StandardScaler
class_weight
train_test_split
XGBClassifier
SS
train_model
(path_to_train_data:str="data/datasets/train.txt",path_save_model:str='data/model/',fasttext_flag:bool=True)
predict_fasttext
(model,path_to_test_data:str="data/datasets/test.txt",training_data:bool=False)
predict
(path_to_test_data:str="data/datasets/test.txt",path_to_model:str='data/model/model_fast.bin',path_to_predictions:str="data/datasets/predictions/",fasttext_flag:bool=True,training_data:bool=False)
path_to_config
config
dataset_path
model_path
predictions_path
fasttext_flag
train_path
test_path
model_name
cat_percent
All occurrences
Please wait...
Page is in error, reload to recover