fix
This commit is contained in:
@@ -14,9 +14,9 @@ warnings.filterwarnings("ignore", category=UserWarning)
|
|||||||
# =========================================================
|
# =========================================================
|
||||||
|
|
||||||
FEATURES_FILENAME = "features.json"
|
FEATURES_FILENAME = "features.json"
|
||||||
MODEL_FILENAME = "model_lgbm.pkl"
|
MODEL_FILENAME = "model.pkl"
|
||||||
SCALER_FILENAME = "scaler_lgbm.pkl"
|
SCALER_FILENAME = "scaler.pkl"
|
||||||
FEATURE_NAMES_FILENAME = "features_lgbm.txt"
|
FEATURE_NAMES_FILENAME = "features.txt"
|
||||||
|
|
||||||
def safe_float(value):
|
def safe_float(value):
|
||||||
if value is None: return 0.0
|
if value is None: return 0.0
|
||||||
|
|||||||
@@ -63,4 +63,5 @@ for (const file of files) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const featurePath = path.join(workingDir, values.fileName ?? 'features.json');
|
const featurePath = path.join(workingDir, values.fileName ?? 'features.json');
|
||||||
fs.writeFileSync(featurePath, JSON.stringify(features, null, 2), 'utf-8');
|
fs.writeFileSync(featurePath, JSON.stringify(features, null, 2), 'utf-8');
|
||||||
|
console.log(`Successfully saved factors to ${featurePath}`);
|
||||||
|
|||||||
Reference in New Issue
Block a user