def twins(start_no=1):
    corp = Corp()
    corps = corp.get_eval_corps()

    params = TrainParams("EACH")
    params.result_file_name = 'twins_result'

    invests = LearningNMockInvestment(params)
    invests.let_train_invests_twins(corps, start_no)
def top10_model():
    corp = Corp()
    corps = corp.get_eval_corps()

    params = TrainParams()
    params.invest_type = 'top10'
    params.result_file_name = "top10_result"
    invests = LearningNMockTop10(params)
    invests.let_train_invests_top10(corps)
def top10_model(train_model='rnn'):
    """상위10개를 가지고 투자하는 모델"""
    corp = Corp()
    corps = corp.get_eval_corps()

    params = GlobalParams(train_model=train_model)
    params.invest_type = 'top10'
    params.result_file_name = "top10_result"
    invests = LearningNMockTop10(params)
    invests.train_n_invests_top10(corps)
def get_corps():
    corp = Corp()
    return corp.get_eval_corps()