|
1 | 1 | using DataFrames, InteractiveFixedEffectModels,Test, CSV
|
2 |
| -df = df = DataFrame(CSV.File(joinpath(dirname(pathof(InteractiveFixedEffectModels)), "../dataset/Cigar.csv"))) |
| 2 | +df = DataFrame(CSV.File(joinpath(dirname(pathof(InteractiveFixedEffectModels)), "../dataset/Cigar.csv"))) |
3 | 3 |
|
4 | 4 | for method in [:gauss_seidel, :dogleg, :levenberg_marquardt]
|
5 | 5 | println(method)
|
6 | 6 | model = @formula Sales ~ 0 + ife(State, Year, 1)
|
7 |
| - result = regife(df, model, method = method, maxiter = 10_000) |
| 7 | + result = regife(df, model, method = method, maxiter = 10_000, save = true) |
8 | 8 | @test abs(result.augmentdf[1, :factors1]) ≈ 0.18662770198472406 atol = 1e-1
|
9 | 9 | @test abs(result.augmentdf[1, :loadings1]) ≈ 587.2272 atol = 1e-1
|
10 | 10 | @test result.augmentdf[1, :residuals] ≈ - 15.6928 atol = 1e-1
|
11 | 11 |
|
12 | 12 | model = @formula Sales ~ 0 + ife(State, Year, 2)
|
13 |
| - result = regife(df, model, method = method, maxiter = 10_000) |
| 13 | + result = regife(df, model, method = method, maxiter = 10_000, save = true) |
14 | 14 | @test abs(result.augmentdf[1, :factors1]) ≈ 0.18662770198472406 atol = 1e-1
|
15 | 15 | @test abs(result.augmentdf[1, :loadings1]) ≈ 587.227 atol = 1e-1
|
16 | 16 | @test result.augmentdf[1, :residuals] ≈ 2.16611 atol = 1e-1
|
17 | 17 |
|
18 | 18 | model = @formula Sales ~ ife(State, Year, 1) + fe(State)
|
19 |
| - result = regife(df, model, method = method, maxiter = 10_000) |
| 19 | + result = regife(df, model, method = method, maxiter = 10_000, save = true) |
20 | 20 | @test abs(result.augmentdf[1, :factors1]) ≈ 0.17636 atol = 1e-1
|
21 | 21 | @test abs(result.augmentdf[1, :loadings1]) ≈ 20.176432452716522 atol = 1e-1
|
22 | 22 | @test result.augmentdf[1, :residuals] ≈ - 10.0181 atol = 1e-1
|
23 | 23 | @test result.augmentdf[1, :fe_State] ≈ 107.4766 atol = 1e-1
|
24 | 24 |
|
25 | 25 | model = @formula Sales ~ ife(State, Year, 2) + fe(State)
|
26 |
| - result = regife(df, model, method = method, maxiter = 10_000) |
| 26 | + result = regife(df, model, method = method, maxiter = 10_000, save = true) |
27 | 27 | @test abs(result.augmentdf[1, :factors2]) ≈ 0.244 atol = 1e-1
|
28 | 28 | @test abs(result.augmentdf[1, :loadings2]) ≈ 49.7943 atol = 1e-1
|
29 | 29 | @test result.augmentdf[1, :residuals] ≈ 2.165319 atol = 1e-1
|
30 | 30 | @test result.augmentdf[1, :fe_State] ≈ 107.47666 atol = 1e-1
|
31 | 31 |
|
32 | 32 | model = @formula Sales ~ ife(State, Year, 2) + fe(State)
|
33 |
| - result = regife(df, model, subset = df.State .<= 30, method = method, maxiter = 10_000) |
| 33 | + result = regife(df, model, subset = df.State .<= 30, method = method, maxiter = 10_000, save = true) |
34 | 34 | @test size(result.augmentdf, 1) == size(df, 1)
|
35 | 35 | @test abs(result.augmentdf[1, :factors1]) ≈ 0.20215 atol = 1e-1
|
36 | 36 | @test abs(result.augmentdf[1, :loadings1]) ≈ 29.546 atol = 1e-1
|
|
0 commit comments