from src.experiment_plot import experiment
from src.house_invest_solution import HouseInvestSolution

'''
You want to learn the investment return details, like net income, yield rate and etc.
If you only care investment plan/solution independently, uncomment "treatment_solution=None". Set the parameters 
in control_solution, then run the code.
If you want to learn by comparison, set parameters in both control and treatment solutions, then run the code.
'''
control_solution = HouseInvestSolution(price=500000, down_payment_ratio=0.2, mortgage_yr=30, interest_rate_yr=0.03,
                                       hoa_month=200, maintain_yr=1000, property_tax_yr=5000, rent_month=4000,
                                       house_price_change_yr=0.01)

treatment_solution = HouseInvestSolution(price=500000, down_payment_ratio=1, mortgage_yr=30, interest_rate_yr=0.03,
                                         hoa_month=200, maintain_yr=1000, property_tax_yr=5000, rent_month=4000,
                                         house_price_change_yr=0.01)
# treatment_solution = None

experiment(control=control_solution, treatment=treatment_solution, target_yr=30, output='img', debug=False)
print("Done")
6946b376876a5d07b0b9e4c53a25050242f1b3a0
<mask token>

<mask token>
experiment(control=control_solution, treatment=treatment_solution,
    target_yr=30, output='img', debug=False)
print('Done')

<mask token>
control_solution = HouseInvestSolution(price=500000, down_payment_ratio=0.2,
    mortgage_yr=30, interest_rate_yr=0.03, hoa_month=200, maintain_yr=1000,
    property_tax_yr=5000, rent_month=4000, house_price_change_yr=0.01)
treatment_solution = HouseInvestSolution(price=500000, down_payment_ratio=1,
    mortgage_yr=30, interest_rate_yr=0.03, hoa_month=200, maintain_yr=1000,
    property_tax_yr=5000, rent_month=4000, house_price_change_yr=0.01)
experiment(control=control_solution, treatment=treatment_solution,
    target_yr=30, output='img', debug=False)
print('Done')

from src.experiment_plot import experiment
from src.house_invest_solution import HouseInvestSolution
<mask token>
control_solution = HouseInvestSolution(price=500000, down_payment_ratio=0.2,
    mortgage_yr=30, interest_rate_yr=0.03, hoa_month=200, maintain_yr=1000,
    property_tax_yr=5000, rent_month=4000, house_price_change_yr=0.01)
treatment_solution = HouseInvestSolution(price=500000, down_payment_ratio=1,
    mortgage_yr=30, interest_rate_yr=0.03, hoa_month=200, maintain_yr=1000,
    property_tax_yr=5000, rent_month=4000, house_price_change_yr=0.01)
experiment(control=control_solution, treatment=treatment_solution,
    target_yr=30, output='img', debug=False)
print('Done')

from src.experiment_plot import experiment
from src.house_invest_solution import HouseInvestSolution

'''
You want to learn the investment return details, like net income, yield rate and etc.
If you only care investment plan/solution independently, uncomment "treatment_solution=None". Set the parameters 
in control_solution, then run the code.
If you want to learn by comparison, set parameters in both control and treatment solutions, then run the code.
'''
control_solution = HouseInvestSolution(price=500000, down_payment_ratio=0.2, mortgage_yr=30, interest_rate_yr=0.03,
                                       hoa_month=200, maintain_yr=1000, property_tax_yr=5000, rent_month=4000,
                                       house_price_change_yr=0.01)

treatment_solution = HouseInvestSolution(price=500000, down_payment_ratio=1, mortgage_yr=30, interest_rate_yr=0.03,
                                         hoa_month=200, maintain_yr=1000, property_tax_yr=5000, rent_month=4000,
                                         house_price_change_yr=0.01)
# treatment_solution = None

experiment(control=control_solution, treatment=treatment_solution, target_yr=30, output='img', debug=False)
print("Done")