File size: 1,590 Bytes
2ec1956
72739b7
 
b51800e
 
72739b7
b51800e
72739b7
 
b51800e
 
 
 
f6a1b4e
 
72739b7
 
 
 
87b6f53
2ec1956
 
 
b51800e
 
 
2ec1956
b51800e
2ec1956
b51800e
2ec1956
a92a468
 
 
2ec1956
b51800e
2ec1956
a92a468
 
2ec1956
a92a468
 
 
 
 
 
 
 
 
 
 
b51800e
 
 
 
 
 
 
 
2ec1956
b51800e
 
2ec1956
a92a468
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
---
datasets:
- BasStein/250000-randomfunctions-2d
language:
- en
library_name: tf-keras
license: apache-2.0
metrics:
- mse
tags:
- doe2vec
- exploratory-landscape-analysis
- autoencoders
co2_eq_emissions:
  emissions: 0.0363
  source: code carbon
  training_type: pre-training
  geographical_location: Leiden, The Netherlands
  hardware_used: 1 Tesla T4
---

## Model description

DoE2Vec model that can transform any design of experiments (function landscape) to a feature vector.  
For different input dimensions or sample size you require a different model.  
Each model name is build up like doe2vec-d{dimension\}-m{sample size}-ls{latent size}-{AE or VAE}-kl{Kl loss weight}

Example code of loading this huggingface model using the doe2vec package.

First install the package

```zsh
pip install doe2vec
```

Then import and load the model.

```python
from doe2vec import doe_model

obj = doe_model(
    2,
    8,
    latent_dim=24,
    kl_weight=0.001,
    model_type="VAE"
)
obj.load_from_huggingface()
#test the model
obj.plot_label_clusters_bbob()
```

## Intended uses & limitations

The model is intended to be used to generate feature representations for optimization function landscapes.
The representations can then be used for downstream tasks such as automatic optimization pipelines and meta-learning.


## Training procedure

The model is trained using a weighed KL loss and mean squared error reconstruction loss.
The model is trained using 250.000 randomly generated functions (see the dataset) over 100 epochs.

- **Hardware:** 1x Tesla T4 GPU
- **Optimizer:** Adam