Построение модели с сохранением файла размером 1 кб

Здравствуйте программисты. Надеюсь, у вас все хорошо. Заранее спасибо Проблема: Когда я создаю свой проект,

# build models for project with id project_id
def build_id(project_id):
    # get directory path to store models in
    path = fetch_model_path(project_id, True)
    # train model
    model, scaler_in, scaler_out = train_project_models(project_id)
    # ensure model was trained
    if model is None:
        return False
    # store models
    store_model(f'{path}/model.pkl', model)
    store_model(f'{path}/scaler_in.pkl', scaler_in)
    store_model(f'{path}/scaler_out.pkl', scaler_out)
    # clear current loaded model from memory
    keras_clear()
    return True

сборка проекта прошла успешно, но сохранился файл размером 1 кб. Если уже существует файл определенного размера, модель уменьшит размер файла file_size до 1kb

получить путь к моделям для проекта с id project_id

def fetch_model_path(project_id, make_dirs=False):
    # get directory path for project_id
    path = JaProjectManagement.objects.all().get(
        id=project_id).url

    # do necessary transformations
    path = path if path.find('://') == -1 else path.split('://')[1]
    path = str(project_id) + '_' + path.replace('.', '_').replace('/', '')
    path = os.path.join(MODEL_PATH, path)


    # make directories if asked
    if make_dirs:
        os.makedirs(path, exist_ok=True)

    return path 

Модель магазина

# dump model to path
def store_model(path, model):
    with open(path, 'wb') as f:
            model_file = File(f)
            pickle.dump(path, model_file)
               
   

Вернуться на верх