exo/lm-dataset-preparer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Configurable model categories;

Browse Source
This commit is contained in:
extio-de
2024-12-20 00:14:02 +01:00
parent 74eaa5c72b
commit d44fd53323
9 changed files with 79 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main/java/de/extio/lmdatasetprep/tools/DatasetTool.java
View File

@@ -3,6 +3,22 @@ package de.extio.lmdatasetprep.tools;
import java.util.Properties;
import java.util.function.Consumer;
import de.extio.lmlib.client.Client;
import de.extio.lmlib.client.ClientService;
import de.extio.lmlib.profile.ModelCategory;
public interface DatasetTool extends Consumer<Properties> {
default String getModelCategoryPropertyName() {
return null;
}
default ModelCategory getModelCategory(final Properties properties) {
return ModelCategory.valueOf(properties.getProperty(this.getModelCategoryPropertyName()));
}
default Client getClient(final Properties properties, final ClientService clientService) {
return clientService.getClient(this.getModelCategory(properties));
}
}