Model-CC-Space / config.py
cd14's picture
adding gen ai support
4f60771
raw
history blame
No virus
696 Bytes
'''
Configurations
'''
import os
import argparse
class Config(object):
OPEN_API_KEY='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'
OPENAI_MODEL_TYPE='text-davinci-003'
OPENAI_MODEL_TEMP=0.3
MODELCC_DATA_S3PATH='s3://emailcampaigntrainingdata/modelCC'
TRAINING_DATA_S3PATH='s3://emailcampaigntrainingdata/trainingdata'
MODEL_FILE_NAME='modelCC.sav'
MODEL_ALLOCATION_PREFIX='sagemakermodelcc'
MODEL_BUCKET_NAME='sagemakermodelcc'
DATASET_TRAINING='email_dataset_training'
DATASET_TRAINING_RAW='email_dataset_training_raw'
MODELCC_TRAINING_DATA='training.csv'
MODELCC_TEST_DATA='Xtest.csv'
MODELCC_TEST_LABEL='ytest.csv'
config=Config()